ROCm/jax

Pull Requests Commits

Merge branch 'master' into rename-mastertrace

gnecula committed 5 years ago

Verified d4f5d9f2

Rename master->main in JAX API and internals (#4178)

gnecula committed 5 years ago

Verified 22036372

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

bchetioui committed 5 years ago

Verified 1a87fd3b

applied simple find+sed for 'master' -> 'main'

mattjj committed 5 years ago

76fbb059

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

Fix bug in omnistaging_enabler (#4159)

gnecula committed 5 years ago

Verified 57f49b68

Fix pytype error (#4158)

gnecula committed 5 years ago

Verified 417c9ff7

cleanup: remove duplicate line (#4156)

jakevdp committed 5 years ago

Verified 29073be0

Use omnistaging env var even when not using absl flags for config. (#4152)

tomhennigan committed 5 years ago

Verified f0fb7d09

allow random.choice to accept ndarray input (#4145)

mattjj committed 5 years ago

Verified 1d939910

Speed up and clean up geomspace test. (#4149)

hawkinsp committed 5 years ago

Verified 01319fb6

Delete batching.last. (#4148)

hawkinsp committed 5 years ago

Verified 4bf3d6e9

Remove workaround for illegal vmap out_axes. (#4147)

hawkinsp committed 5 years ago

Verified 8c8060e1

Do not call asarray() on inputs of jax.random.choice (#4137)

jakevdp committed 5 years ago

Verified 6d54eb56

Rename collectives into "collective operations" for the pmap function. (#4136)

jblespiau committed 5 years ago

Verified f959219a

make pe.abstract_eval_fun use omnistaging (#4139)

mattjj committed 5 years ago

Verified f4b05bc9

Merge pull request #4140 from sharadmv/patch-2

mattjj committed 5 years ago

Verified 04173b33

Remove frame check assertion in `extend_axis_env`.

sharadmv committed 5 years ago

774b5f68

Add support for negative axes to vmap. (#4111)

hawkinsp committed 5 years ago

Verified e06a6ab6

Fix scan carry types in gradient of complex ODE (#4130)

dav0dea committed 5 years ago

Verified 603f0c1e

Older