ROCm/jax

Pull Requests Commits

fuse lazy expressions into eager computations

mattjj committed 6 years ago

2d2958c9

mattjj committed 6 years ago

b04019ea

implement lazy sublanguage

mattjj committed 6 years ago

ad9b6d4d

Implement batched cholesky decomposition using LAPACK/Cusolver (#1956)

hawkinsp committed 6 years ago

Verified c5a9eba3

make control flow abstract eval to shaped level

mattjj committed 6 years ago

7da75587

Use numpy function directly instead of copying source code

Clemens Schmid committed 6 years ago

0c9aacf1

Add np.iterable

Clemens Schmid committed 6 years ago

58ee0a8e

Allow Var to be a key in a JAX tree.

tomhennigan committed 6 years ago

e2f7f37f

added fftfreq, corresponding tests, and documentation links.

archis committed 6 years ago

1e8c9384

lower away grad_and_aux pure jaxpr tracers

mattjj committed 6 years ago

80d8ee4e

tweak shape error message, add test

mattjj committed 6 years ago

bb9cd233

Concatenate error messages under numpy.{zeros,ones,full}. Closes #1822

froystig committed 6 years ago

1ca9e9b2

Merge pull request #1947 from gnecula/cleanup

gnecula committed 6 years ago

Verified 29db4203

Attempt a fix for PY2

gnecula committed 6 years ago

4b2a5a1f

Merge pull request #1946 from gnecula/bug_fix

gnecula committed 6 years ago

Verified 5465b9e8

Removed unused imports

gnecula committed 6 years ago

df374fa3

Added some more documentation to the linear_util module

gnecula committed 6 years ago

528a69f3

Fix bug with caching in presence of JVP and JIT

gnecula committed 6 years ago

ea9e9328

added fft2 and ifft2, corresponding tests, and documentation links. (#1939)

joglekara committed 6 years ago

ca155129

bump version for pypi

mattjj committed 6 years ago

2c87f72a

Don't serialize jaxpr equations into XLA metadata.

skye committed 6 years ago

52cafb37

fft and ifft implementation (#1926)

joglekara committed 6 years ago

d9c6a5f4

Fix minor typo in Common_Gotchas_in_JAX.ipynb

johnmellor committed 6 years ago

fd606747

add faster reshape utility function

mattjj committed 6 years ago

b380ac1f

Fix for compatibility with python>=3.7 and numpy>=1.18

neerajprad committed 6 years ago

ed309b6f

add tests for #1640, adapt make_jaxpr staging

mattjj committed 6 years ago

82dbf913

Update docs/conf.py

Yo committed 6 years ago

322ebe7c

flowed committed 6 years ago

e0693fe6

fix error in autodiff cookbook: 3x not 2x

mattjj committed 6 years ago

Verified f5723848

fix typo in autodiff cookbook (#1921)

dbieber committed 6 years ago

30bede1f

Older