ROCm/jax

Pull Requests Commits

mattjj committed 5 years ago

4a256212

mattjj committed 5 years ago

e323a7c0

mattjj committed 5 years ago

b71e31d4

Add reverse flag in associative scan (#4181)

AdrienCorenflos committed 5 years ago

Verified 96278e67

[jax2tf] Generator for the documentation of operations with limited support (WIP) (#4193)

bchetioui committed 5 years ago

Verified bcf9777b

[jax2tf] Flip the with_gradient=True; was flipped back by mistake (#4200)

gnecula committed 5 years ago

Verified abdd1388

[jax2tf] Implementation of random_gamma (#4192)

gnecula committed 5 years ago

Verified 5eac4772

Add jax.numpy.array_split (#4197)

ajcr committed 5 years ago

Verified 708d07d5

better jax.numpy.tile implementation (#4190)

mattjj committed 5 years ago

Verified 04f9a7e5

copysign: promote to inexact to match numpy & support unsigned inputs (#4188)

jakevdp committed 5 years ago

Verified 421550a9

[jax2tf] Indicate the version of TF used in tests in README. (#4185)

bchetioui committed 5 years ago

Verified 0cdb1f7e

Add more features to the C++ jax.jit. (#4169)

jblespiau committed 5 years ago

Verified bdd65453

jnp.abs(): support boolean inputs (#4186)

jakevdp committed 5 years ago

Verified 36368a2a

Fix axis checking and remove extra print statement (#4184)

hamzamerzic committed 5 years ago

Verified 44bcf7e7

[jax2tf] Turn on with_gradient by default (#4180)

gnecula committed 5 years ago

Verified b6b1f5e3

More renaming of master to main in JAX internals (#4179)

gnecula committed 5 years ago

Verified 634c6259

lax.associative_scan: fix docstring examples (#4172)

jakevdp committed 5 years ago

Verified ffbfadd8

applied simple find+sed for 'master' -> 'main' (#4174)

mattjj committed 5 years ago

Verified 6b6789a5

Implement a proper shape checking rule for gather. (#4166)

bchetioui committed 5 years ago

Verified 1a87fd3b

Add support for axis_index inside vmap (#4168)

apaszke committed 5 years ago

Verified a33f4dd8

Avoid calling jnp.sum() on list (#4163)

jakevdp committed 5 years ago

Verified 1dab791a

Addition of one more conclusive polynomial comparison case. (#4167)

bchetioui committed 5 years ago

Verified 04f9ff7f

Add support for binding axis_name in gmap

apaszke committed 5 years ago

7210d6f5

Add benchmarks for specifically the dispatch time. (#4128)

jblespiau committed 5 years ago

Verified e95d5701

Revert "Delete batching.last. (#4148)" (#4160)

gnecula committed 5 years ago

Verified 36846e0e

[jax2tf] Added conversion for scatter*_p primitives. (#4091)

bchetioui committed 5 years ago

Verified a7faf090

Implement a proper shape checking rule for scatter. (#4144)

bchetioui committed 5 years ago

Verified 4d7396aa

Add a boolean to _check_shapelike to accept or reject shapes (#4108)

bchetioui committed 5 years ago

Verified 80114e51

[jax2tf] Add testing for add/mul/min/max conversion. (#4142)

bchetioui committed 5 years ago

Verified 1dc71b2f

Revert "Increase tolerance for CPU test LaxBackedNumpyTests::testCorrCoef (#4080)" (#4151)

gnecula committed 5 years ago

Verified c76b84f6

Older