ROCm/jax

Pull Requests Commits

Cleanup: use _canonicalize_axis() utility where possible (#4270)

jakevdp committed 5 years ago

Verified 83b4f3b9

Move failing CPPJitTest test case to PythonJitTest (#4268)

skye committed 5 years ago

Verified ee9dccf3

Pin pygments version in RTD build. (#4267)

skye committed 5 years ago

Verified 6dc161cf

Fixing weird behavior in segment_sum when num_segments is None (#4034)

alvarosg committed 5 years ago

Verified ca1d8f41

Call check_arraylike() in jax.numpy reductions (#4195)

jakevdp committed 5 years ago

Verified 3b7329c9

Extend axis env while translating the pmapped jaxpr to XLA

apaszke committed 5 years ago

40fb01b4

implement jnp.apply_along_axis (#4253)

jakevdp committed 5 years ago

Verified 8a18b10f

Update README to point to jaxlib-0.1.55. (#4256)

zhangqiaorjc committed 5 years ago

Verified 7e694bd3

Clarify when jaxlib version should be bumped. (#4250)

skye committed 5 years ago

Verified fc39332c

shoyer committed 5 years ago

f9b358b9

Don't support integer inputs; better docstring

shoyer committed 5 years ago

71906e38

fix documentation typo (#4252)

jakevdp committed 5 years ago

Verified 2a33b3d3

Merge branch 'master' into linear-transpose

shoyer committed 5 years ago

91f757e2

Bump TF hash to get an upstream LLVM GCC fix. (#4251)

zhangqiaorjc committed 5 years ago

Verified 82af356b

Change lax_linalg.lu to return a permutation representation of the partial pivoting information. (#4241)

hawkinsp committed 5 years ago

Verified cf65f6b2

Revert "Revert "Delete batching.last. (#4148)" (#4160)" (#4242)

hawkinsp committed 5 years ago

Verified b67e42a3

Add comments for residuals from f_bwd. (#4244)

zhangqiaorjc committed 5 years ago

Verified 26a53ae5

[jax2tf] Fix test failure on TPUs (#4247)

gnecula committed 5 years ago

Verified 0962ceb0

[jax2tf] Cleanup test_unary_elementwise. (#4246)

bchetioui committed 5 years ago

Verified 29f97afa

[jax2tf] Moved the limitations for XlaSort to correctness_stats (#4237)

gnecula committed 5 years ago

Verified f0a3fd4a

Reorder nocuda/cuda build to fail early. (#4243)

zhangqiaorjc committed 5 years ago

Verified adb34488

Update install_cuda script to specify cublas. (#4240)

zhangqiaorjc committed 5 years ago

Verified 0b04439f

Update TF dep to a passing commit hash. (#4239)

zhangqiaorjc committed 5 years ago

Verified a14133aa

Interrupt lu transformation generators whenever an exception occurs

apaszke committed 5 years ago

3f8aaabb

[jax2tf] Add a template file for documentation generation. (#4219)

bchetioui committed 5 years ago

Verified 70891f46

[jax2tf] Clean up test_dynamic_slice. (#4236)

bchetioui committed 5 years ago

Verified 053cd5aa

Add axis_index_groups support to all_gather. (#4194)

inoryy committed 5 years ago

Verified bff24bdd

[jax2tf] Updated test_pad to test all dtypes and remove old (#4235)

bchetioui committed 5 years ago

Verified f908f6f2

[jax2tf] Clean up code for XlaGather, experimental_compile not necessary (#4030)

gnecula committed 5 years ago

Verified ee38e716

Fix Dockerfile wheel installation issues. (#4232)

zhangqiaorjc committed 5 years ago

Verified 3cf73367

Newer Older