ROCm/jax

Pull Requests Commits

added test skips based on jaxlib version

mattjj committed 5 years ago

4a19de60

[JAX] Add support for retaining ownership of DLPack tensors.

hawkinsp committed 5 years ago

d001ac6b

Merge pull request #4657 from skye:jaxlib2

a-googler committed 5 years ago

41eea2c7

Update README and jaxlib.__version__ for new jaxlib release

skye committed 5 years ago

a0821696

Merge pull request #4656 from hawkinsp:xla

a-googler committed 5 years ago

52dbeeee

hawkinsp committed 5 years ago

29a83ddd

Merge pull request #4655 from skye:cpp_jit

a-googler committed 5 years ago

9c0452d7

Don't turn on C++ jit by default until we're sure it works everywhere.

skye committed 5 years ago

8ef49a3b

Copybara import of the project:

axch committed 5 years ago

537427aa

Merge pull request #4652 from hawkinsp:stats

a-googler committed 5 years ago

455f6bd0

Add missing jax.scipy.stats distributions to the docs.

hawkinsp committed 5 years ago

c0b480bd

Merge pull request #4647 from google:update-pypi

a-googler committed 5 years ago

4b8334ab

update version and changelog for pypi

mattjj committed 5 years ago

87623ead

Merge pull request #4571 from skye:warning

a-googler committed 5 years ago

e567b22c

Merge pull request #4580 from skye:jaxlib

a-googler committed 5 years ago

60141d95

Merge pull request #4641 from jakevdp:indexing-deprecation

a-googler committed 5 years ago

6a54f167

Merge pull request #4642 from lmmx:master

a-googler committed 5 years ago

c17fbe23

Add deprecation warning for indexing with non-tuple sequences

jakevdp committed 5 years ago

c56f76db

A couple of typo/gap fixes in PRNG design notes

lmmx committed 5 years ago

11e6f49c

Merge pull request #4616 from bchetioui:test_linear_solve

a-googler committed 5 years ago

b7ec636c

[jax2tf] Implemented a conversion for linear_solve_p

bchetioui committed 5 years ago

a8cfc748

Merge pull request #4613 from bchetioui:test_lu

a-googler committed 5 years ago

f785794a

[jax2tf] Implemented conversion of lu_p using _convert_jax_impl.

bchetioui committed 5 years ago

7d301877

Merge pull request #4634 from hawkinsp:src2

a-googler committed 5 years ago

26771b31

Merge pull request #4633 from hawkinsp:src

a-googler committed 5 years ago

e5e33f57

Move implementation of jax.lax into jax._src.lax.

hawkinsp committed 5 years ago

10b7d7d7

Import jax.nn.functions by default to fix breakage.

hawkinsp committed 5 years ago

b0784835

Move jax.third_party to jax._src.third_party.

hawkinsp committed 5 years ago

3ddd3905

Merge pull request #4631 from hawkinsp:src

a-googler committed 5 years ago

09dfd474

Move jax.nn implementation into jax._src.nn.

hawkinsp committed 5 years ago

b2808dc8

Older