ROCm/jax

Pull Requests Commits

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

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

Copybara import of the project:

a-googler committed 5 years ago

e9909ce0

Merge pull request #4628 from hawkinsp:src3

a-googler committed 5 years ago

15bd7f64

Merge pull request #4627 from hawkinsp:src2

a-googler committed 5 years ago

8841dd4e

Newer Older