ROCm/jax

Pull Requests Commits

initial unrolled and block-unrolled scan

froystig committed 5 years ago

59bf9af7

fix disable_jit logic in lax.cond and lax.while_loop (#3156)

mattjj committed 5 years ago

Verified 42b425d8

Add lax implementation of np.isin() and np.in1d() (#3087)

jakevdp committed 5 years ago

Verified 3141ff83

improve pmap unbound axis error, fixes #3120 (#3152)

mattjj committed 5 years ago

Verified ccb203c8

fix broken TPU test (#3153)

jakevdp committed 5 years ago

Verified b7ff305a

make lax.psum promote bool -> int (#3150)

mattjj committed 5 years ago

Verified bc47a32c

add erf and erfc rules (#3051)

jacobjinkelly committed 5 years ago

Verified 83a339e1

Correct calculation of loss and increase learning rate (#3113)

Baschdl committed 5 years ago

Verified b6777c07

improve errors for complex derivs, fixes #3121 (#3149)

mattjj committed 5 years ago

Verified 850f1afd

Expand type support for random uniform() & randint() (#3138)

jakevdp committed 5 years ago

Verified 8fe26190

Exported lax from jax/__init__.py (#3135)

superbobry committed 5 years ago

Verified 73b76e99

Fix indentation for docstrings in jax.experimental.host_callback (#3119)

shoyer committed 5 years ago

Verified 77e31323

Add gradients to the scatter_max and scatter_min operations. (#3111)

alexdavies committed 5 years ago

Verified 85fe5a28

Fix a corner case in `repeat`. (#3117)

midfield committed 5 years ago

Verified 8d0749f1

Improve sharded_jit error message and fix test (#3145)

skye committed 5 years ago

Verified d53bab93

Fix in pxla._inner_partitions (#3146)

skye committed 5 years ago

Verified 083cdd38

Move some sharded_jit functionality into pxla.py (#3142)

skye committed 5 years ago

Verified cb77f2a2

Add with_sharding_constraint method to be used within sharded_jit. (#3100)

skye committed 5 years ago

Verified 9152b760

Fix gradient for jax.scipy.ndimage.map_coordinates (#3110)

shoyer committed 5 years ago

Verified 46034329

Add a primitive integer_pow() for values raised to a fixed integer scalar. (#3140)

hawkinsp committed 5 years ago

Verified 36e7fad1

Added link to README (#3139)

sursu committed 5 years ago

Verified a9c1b386

Add sharded_jit translation rule. (#3099)

skye committed 5 years ago

Verified ed0e227e

Remove some uses of jax.partial. (#3131)

hawkinsp committed 5 years ago

Verified bc5a0b33

Fixed link in FAQ (#3129)

gnecula committed 5 years ago

Verified b071b12a

Remove redundant flag configuration from lax_scipy_sparse_test (#3128)

shoyer committed 5 years ago

Verified 2be471cc

Implement np.histogram_bin_edges and np.histogram (#3081)

jakevdp committed 5 years ago

Verified 912b2825

Test code in docs and api.py docstrings (#2994)

Jamie Townsend committed 5 years ago

Verified 670fab59

Fix documentation for `nn.elu`, `nn.celu`, and `lax.expm1`. (#3116)

schmrlng committed 5 years ago

Verified 510af1de

Add support for 8- and 16-bit output in _random_bits (#3090)

jakevdp committed 5 years ago

Verified e675f804

Update random.logistic() to prevent infinities (#3048)

jakevdp committed 5 years ago

Verified 12a9af86

Older