ROCm/jax

Pull Requests Commits

Fix broadcasting in random.uniform and randint. (#4035)

ethanluoyc committed 5 years ago

Verified 6e4ec7cb

Enable s8/u8/s16/u16 types on TPU in tests. (#4032)

hawkinsp committed 5 years ago

Verified a169743f

[host_callback] Add support for custom gradients inside control-flow (#4029)

gnecula committed 5 years ago

Verified ece05d28

skip more ode tests on gpu, b/c slow to compile (#4028)

mattjj committed 5 years ago

Verified c564aca7

Fix type promotion in np.clip (#4023)

jakevdp committed 5 years ago

Verified 5fcca184

cumulative jet rules (#4000)

thisiscam committed 5 years ago

Verified fe9f264b

Fix typos in api.py docstrings (#4021)

j-towns committed 5 years ago

Verified 1e077129

Implement the invertible decorator in terms of custom_vjp (#3957)

apaszke committed 5 years ago

Verified cd54dd97

[jax2tf] Disable the CI tests for jax2tf. (#4019)

gnecula committed 5 years ago

Verified 0b99ca89

[jax2tf] Change the tests that assert failure in TF (#4018)

gnecula committed 5 years ago

Verified ebc5e8bf

Restrict JAX2TF tests to cases supported by TF2 ops. (#4010)

cheshire committed 5 years ago

Verified 55913163

Remove type restrictions (#4017)

majnemer committed 5 years ago

Verified d100327b

use fewer internal APIs in custom interp notebook (#4016)

mattjj committed 5 years ago

Verified 09d8ac14

Remove type restrictions (#4011)

majnemer committed 5 years ago

Verified 265c3faa

make make_jaxpr work on tracer example args (#4014)

mattjj committed 5 years ago

Verified 6a3b9205

Remove type restrictions (#4009)

majnemer committed 5 years ago

Verified eb086f9b

Add Bessel functions in jax.numpy & jax.scipy.special (#4007)

jakevdp committed 5 years ago

Verified d551cec6

Fix warnings in pmap_test.py (#3977)

j-towns committed 5 years ago

Verified 89e8a088

Add jax.numpy.invert (#3998)

jakevdp committed 5 years ago

Verified dd304f62

add helper for flax to be omnistaging-compatible (#4004)

mattjj committed 5 years ago

Verified d46ea969

Fixes and tests for jax.random.multivariate_normal (#4002)

slinderman committed 5 years ago

Verified ea88c55f

Improve type annotations for `jit` and `vmap`. (#3938)

aslanides committed 5 years ago

Verified 038c85da

Update docs requirements. (#3969)

jlebar committed 5 years ago

Verified fb02cc80

github actions: cache pypi dependencies (#3990)

jakevdp committed 5 years ago

Verified 481b6f15

Add nbytes property to jax.numpy arrays. (#3988)

jakevdp committed 5 years ago

Verified da51f73e

[jax2tf] implementation of select_and_gather_add conversion. (#3987)

bchetioui committed 5 years ago

Verified b756f294

[jax2tf] Another attempt to work around the nested XLA compilation bug (#3965)

gnecula committed 5 years ago

Verified e39420b5

[jax2tf] First draft of SVD conversion. (#3966)

bchetioui committed 5 years ago

Verified 4af7d2a4

Add missing functions to jax.numpy docs (#3981)

jlebar committed 5 years ago

Verified a98103f6

Fix jax.checkpoint in API docs (#3980)

shoyer committed 5 years ago

Verified 816bcd71

Newer Older