ROCm/jax

Pull Requests Commits

Fix build.py to work on Linux once again.

hawkinsp committed 5 years ago

4bb5dca7

[jax2tf] Added testing code for div and rem.

bchetioui committed 5 years ago

7c471b6b

[JAX] Move contents of jax.random into jax._src.random.

hawkinsp committed 5 years ago

77827538

Deprecation: Raise a TypeError when indexing with a non-tuple sequence

jakevdp committed 5 years ago

b6def8ca

Add .bazelversion file.

hawkinsp committed 5 years ago

8b56501d

Merge pull request #4843 from cloudhan:winbuild

a-googler committed 5 years ago

f0cddca8

Remove jax.experimental.optix.

hawkinsp committed 5 years ago

195e13c1

[JAX] Delete jax.lax_linalg.

hawkinsp committed 5 years ago

783a27fa

Build on Windows

cloudhan committed 5 years ago

a6acce58

Temporarily disable JAX tests that are failing due to LLVM commit https://github.com/llvm/llvm-project/commit/fcad8d3635cff61a2749dcef94c0d51fa1e3e413

hawkinsp committed 5 years ago

f0e22b40

[JAX] Move pprint_util into jax._src.

hawkinsp committed 5 years ago

84c723fc

[JAX] Delete jax.source_info_util.

hawkinsp committed 5 years ago

3a1c9210

[JAX] Move implementation of jax.scipy.sparse.linalg into jax._src.

hawkinsp committed 5 years ago

94cd2046

Remove devices from xla.AxisEnv

apaszke committed 5 years ago

b55b44d3

Merge pull request #4746 from skye:pod_sharded_jit

a-googler committed 5 years ago

c2b01f11

Merge pull request #4947 from jakevdp:new-changelog

a-googler committed 5 years ago

2d47124a

Add initial changelog for jax v0.2.7

jakevdp committed 5 years ago

2f03d70a

Merge pull request #4946 from jakevdp:jax-026

a-googler committed 5 years ago

0978ccbd

Merge pull request #4926 from bchetioui:test_transpose

a-googler committed 5 years ago

050b8795

Add multi-host support to sharded_jit.

skye committed 5 years ago

0b4d3e3f

update version & CHANGELOG for jax v0.2.6

jakevdp committed 5 years ago

ef4adb5c

Merge pull request #4868 from sharadmv:variadic_reduce

a-googler committed 5 years ago

5afc7758

Merge pull request #4940 from jakevdp:var-warning

a-googler committed 5 years ago

3c671e86

Merge pull request #4878 from gnecula:revisit_poly

a-googler committed 5 years ago

8cb62f2a

Merge pull request #4944 from hawkinsp:sourceinfo

a-googler committed 5 years ago

3ba7f422

Fix mypy error caused by cyclic import dependency.

hawkinsp committed 5 years ago

13db2c37

Merge pull request #4943 from hawkinsp:sourceinfo

a-googler committed 5 years ago

f939e71c

Use the jax module to determine the jax source directory base, rather than the source_info module.

hawkinsp committed 5 years ago

e38f96dc

Merge pull request #4916 from google:switch-grad-weak-type-test

a-googler committed 5 years ago

8be807b6

Merge pull request #4932 from jakevdp:full-like-shape

a-googler committed 5 years ago

ae63e58b

Newer Older