ROCm/jax

Pull Requests Commits

Merge pull request #2698 from google/issue2107

mattjj committed 5 years ago

Verified ffe229ae

Add missing functions to autodoc

jakevdp committed 5 years ago

4822581a

Implement jax.ops.index_mul. (#2696)

hawkinsp committed 5 years ago

Verified 714b276b

Update XLA. (#2699)

hawkinsp committed 5 years ago

Verified 04de8339

Add implementation of numpy packbits() and unpackbits() (#2695)

jakevdp committed 5 years ago

Verified ea54c066

add custom_jvp for logaddexp / logaddexp2

mattjj committed 5 years ago

2d25773c

Add `--export_dir` and `--baseline_dir` flags to benchmark.py. (#2677)

skye committed 5 years ago

Verified 8c2901cf

Import std::isnan from cmath rather than assuming it is in the global namespace. (#2693)

hawkinsp committed 5 years ago

Verified afefc927

Fix type annotation for uniform.

aslanides committed 5 years ago

6cb69f5c

update version and changelog for pypi

mattjj committed 5 years ago

d8c75da8

Make pytest run over JAX tests warning clean, and error on warnings. (#2674)

hawkinsp committed 5 years ago

Verified 2dc81fb4

Relax tolerance for LaxBackedNumpyTests.testConvolutions

gnecula committed 5 years ago

453dc5f0

Add implementation of np.rollaxis

jakevdp committed 5 years ago

989228a6

Add some types to jax.random and jnp.ndarray.

aslanides committed 5 years ago

c06fe56f

Implement indexing helpers

danieldjohnson committed 5 years ago

98d46d39

Improve error message for indexing into an empty dimension

gnecula committed 5 years ago

f6108674

Relax tolerance for new test on TPU

gnecula committed 5 years ago

c4dd1cf3

Added missing import warnings

gnecula committed 5 years ago

ee2cd449

Merge pull request #2675 from jakevdp/convolve

mattjj committed 5 years ago

Verified 8a95ff09

Add support for mode=same in convolve2d & correlate2d

jakevdp committed 5 years ago

89c9c437

Fix crash for NaN eigenvalues. (#2669)

hawkinsp committed 5 years ago

Verified 414e54aa

fix a typo in docs notebook (#2672)

inoryy committed 5 years ago

Verified 656c3a95

Merge pull request #2673 from google/issue2657

mattjj committed 5 years ago

Verified bad3445d

Add implementations of scipy.signal.convolve & correlate, 1d & 2d

jakevdp committed 5 years ago

edda69ef

add test for #2657

mattjj committed 5 years ago

7750a16c

add missing instantiate_zeros in custom_jvp

mattjj committed 5 years ago

61abdc1e

Remove a workaround for a long-fixed XLA type conversion bug. (#2670)

hawkinsp committed 5 years ago

Verified a3cc9a7d

Update XLA. (#2668)

hawkinsp committed 5 years ago

Verified e904780c

Port np.roots (#2250)

mathisgerdes committed 5 years ago

Verified bd70db79

Add implementations of np.convolve & np.correlate

jakevdp committed 5 years ago

e0ce2545

Newer Older