ROCm/jax

Pull Requests Commits

clean up handling of aux data in jvp_subtrace_aux

mattjj committed 6 years ago

982f8670

fix flakes at head (#3361)

jakevdp committed 6 years ago

Verified 528207bd

Fix polyadd and test (#3344)

ashridh committed 6 years ago

Verified b1adef40

A couple of ad_util.zero were missed in #3222 (#3363)

gnecula committed 6 years ago

Verified 65d95f10

fix a one-character issue from a bad merge (#3362)

mattjj committed 6 years ago

Verified 508821cc

Update optix.py (#3355)

mtthss committed 6 years ago

Verified dc41ff90

Make ad_util.zero a class that carries avals (similar to UndefinedPrimal) (#3222)

apaszke committed 6 years ago

Verified e36c72b9

Edit documentation. (#3358)

hawkinsp committed 6 years ago

Verified 96d72fca

Update README to recommend Windows Subsystem for Linux (WSL) to Windows users. (#3356)

hawkinsp committed 6 years ago

Verified 701bc761

Generate type hints for parameters, even if the parameters aren't documented. (#3337)

hawkinsp committed 6 years ago

Verified 6b948e64

Move tree_call into process_call instead of a primitive

shoyer committed 6 years ago

09d76014

Add (non-working) jacrev example

shoyer committed 6 years ago

44d45dac

Commented-out the literal jaxpr checks in host_callback (#3351)

gnecula committed 6 years ago

Verified 7863f817

make jnp.array faster (#3350)

mattjj committed 6 years ago

Verified fd886b17

WIP test_jacfwd

shoyer committed 6 years ago

657ba226

deflake remainder of jax (#3343)

jakevdp committed 6 years ago

Verified 2a10dbbf

Cleanup: deflake jax.experimental and jax.ops (#3329)

jakevdp committed 6 years ago

Verified fb171723

shoyer committed 6 years ago

58eaf2b5

Don't expand type aliases when generating JAX documentation. (#3336)

hawkinsp committed 6 years ago

Verified 1a3c1805

Fix missing sharded_jit_test import

skye committed 6 years ago

b65abdf5

Handle with_sharding_constraint inside arbitrary subjaxprs. (#3339)

skye committed 6 years ago

Verified 0b5d885e

Add np.polysub (#3319)

ashridh committed 6 years ago

Verified 29740de6

Enable SVD on TPU. (#3334)

hawkinsp committed 6 years ago

Verified 841f21fc

apaszke committed 6 years ago

e2a32f74

Optimize zeros_like_shaped_array

apaszke committed 6 years ago

17472b97

Remove accidental prints

apaszke committed 6 years ago

cfa3b788

Update notebooks

apaszke committed 6 years ago

7a0e8bae

Remove example from ad.instantiate_zeros, fix vmap bug

apaszke committed 6 years ago

3f1d3a73

Fix host_callback

apaszke committed 6 years ago

c5d87073

Make ad_util.zero a class that carries avals (similar to UndefinedPrimal)

apaszke committed 6 years ago

adb442eb

Newer Older