ROCm/jax

Pull Requests Commits

mattjj committed 5 years ago

32ffb247

mattjj committed 5 years ago

d5c94082

fix a custom_jvp vmap bug from @dpfau

mattjj committed 5 years ago

a46d0028

Merge pull request #4663 from google:improve-tracer-error-2

a-googler committed 5 years ago

dcfa7644

only run test with omnistaging on

mattjj committed 5 years ago

2b62a447

mattjj committed 5 years ago

49b1ec22

Merge pull request #4664 from google:4008-fixes

a-googler committed 5 years ago

8684da4c

fixes on #4008 (thanks @apaszke)

mattjj committed 5 years ago

79e3db55

Merge pull request #4662 from google:skip-dlpack-tests-on-jaxlib-version

a-googler committed 5 years ago

00905022

improve an escaped tracer error

mattjj committed 5 years ago

ac6cd23b

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 jax/core.py

gnecula committed 5 years ago

Verified cb591eb7

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

Also convert numpy arrays to hashable values.

jblespiau committed 5 years ago

114c1dd0

Ensure that check_jaxpr is done with abstract values

gnecula committed 5 years ago

09653bf5

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

Newer Older