ROCm/jax

Pull Requests Commits

faster normalized version

mattjj committed 5 years ago

520ba302

first cut at an HMM EM example

mattjj committed 5 years ago

83d540ef

[remat] Change remat lowering to XLA::Conditional (#2391)

trevorcai committed 5 years ago

Verified 620bf430

Allow zero tolerance for jax.test_util.tolerance (#2393)

danieldjohnson committed 5 years ago

Verified 2dfeaeb6

add raises-exception notebook cell metadata (#2402)

mattjj committed 5 years ago

Verified cdf188af

Add jnp vs np out of bounds indexing to Sharp Bits nb (#2378)

hoffmannjordan committed 5 years ago

Verified ffa03403

add register_pytree_node_class, fixes #2396 (#2400)

mattjj committed 5 years ago

Verified cfbdb65a

allow vmap in_axes to be a list, fixes #2367 (#2395)

mattjj committed 5 years ago

Verified ebbcbad5

Replace uses of ExecutePerReplica with ExecuteOnLocalDevices. (#2394)

hawkinsp committed 5 years ago

Verified 9fd69a04

skip new optix test on tpu (cf. #2350)

mattjj committed 5 years ago

cc53aa95

Add a module to apply updates every k steps (and accumulate them otherwise) (#2350)

perolat committed 5 years ago

Verified 5c3b4786

jit lax_numpy.roll (#2392)

shoyer committed 5 years ago

Verified 863576c5

Fix exception causes all over the codebase (#2376)

cool-RR committed 5 years ago

Verified f3f0abb5

Fix ONNX mnist example (#2374)

bddppq committed 5 years ago

Verified ac6a313c

Removed unused imports (#2385)

gnecula committed 5 years ago

Verified c52f32b5

Added some pytype annotations (#2386)

gnecula committed 5 years ago

Verified 282225f6

automatic detection of wheel version (#2373)

JL2718 committed 5 years ago

Verified c53ae2c4

Update lnp -> jnp to fix test failure after merge. (#2389)

hawkinsp committed 5 years ago

Verified 411c8a4b

Implement NumPy sorting routines. (#2318)

us committed 5 years ago

Verified 8339511e

share rng and predicate between all foiled args (#2375)

trevorcai committed 5 years ago

Verified 25dd419c

update gitignore (#2361)

xwinxu committed 5 years ago

Verified a5daafd3

Fix a few type annotations in `api.py`. (#2387)

chr1sj0nes committed 5 years ago

Verified 0080c895

Minor fix to docs which mentioned IOHW where it should be OIHW (#2381)

lucastheis committed 5 years ago

Verified 05e5ccfd

Give `Var`s an `aval`. (#2299)

chr1sj0nes committed 5 years ago

Verified 1e7d13b5

Replace assertEquals with assertEqual (#2384)

gnecula committed 5 years ago

Verified 89dcfe67

Replaced self.assertTrue(...===...) with assertEquals (#2383)

gnecula committed 5 years ago

Verified 04a027b8

Rename np -> jnp and onp -> np in jax/scipy to match the standard JAX… (#2370)

hawkinsp committed 5 years ago

Verified c953ca2d

Rename `lnp` to `jnp` in tests. (#2369)

hawkinsp committed 6 years ago

Verified 61fca0d3

Rename "NeurIPS 2019 JAX demo" to "JAX demo", as it's been presented elsewhere. (#2372)

skye committed 6 years ago

Verified 74faef18

Add pytype annotations to JAX API. (#2368)

hawkinsp committed 6 years ago

Verified 8e251e1f

Older