ROCm/jax

Pull Requests Commits

hoist constants to parameters in jit-compiled functions

froystig committed 5 years ago

182c4622

uses np.prod instead of jnp.prod for shapes (#3236)

joaogui1 committed 5 years ago

Verified e48a4e01

implement np.extract (#3234)

jakevdp committed 5 years ago

Verified 1a43693f

Fix sign error in custom_jvp / custom_vjp. (#3213) (#3219)

jiawen committed 5 years ago

Verified 7c90023d

fix custom_jvp_call_jaxpr transpose function (#3231)

mattjj committed 5 years ago

Verified 572928df

Small cleanup for partial_eval (#3210)

apaszke committed 5 years ago

Verified c1ccbdf1

Fix uses of sets in einsum to avoid nondeterminism. (#3230)

hawkinsp committed 5 years ago

Verified 5475e1db

Add implementation of np.compress (#3227)

jakevdp committed 5 years ago

Verified 9b91e894

Implement np.argwhere & np.flatnonzero (#3223)

jakevdp committed 5 years ago

Verified 5fa6ab51

Update installation directions in README to mention expected CUDA location. (#3190)

skye committed 5 years ago

Verified 6e50aa9b

Fix broadcast_shapes for polymorphic dims (#3216) (#3224)

juliuskunze committed 5 years ago

Verified 02b4fd35

Fix bug in pytype fix. (#3229)

hawkinsp committed 5 years ago

Verified 7d96aae5

mention numpy & scipy convolve functions in gotchas doc. (#3214)

jakevdp committed 5 years ago

Verified 41292a2a

Fix pytype error. (#3226)

hawkinsp committed 5 years ago

Verified 336a0d6e

Relax test tolerance on lax_scipy_test to fix a test failure on Skylake machines at LLVM head. (#3225)

hawkinsp committed 5 years ago

Verified 94b4ccd6

use new gensym in host_callback jaxpr rewriter

froystig committed 5 years ago

c5010cda

update reference jaxprs in tests

froystig committed 5 years ago

da9d8c9c

update core.gensym consumers, address rewrite TODOs in lax control flow rules

froystig committed 5 years ago

1b020fe2

jaxpr-dependent gensym to avoid var duplication

froystig committed 5 years ago

e80e9634

Simplify handling of non-linear equations in backward_pass and fix remat (#3162)

apaszke committed 5 years ago

Verified 8f2d72eb

Added geometric distribution to scipy stats (#3205)

gsp-27 committed 5 years ago

Verified ec3b593c

Adjust complex64 tolerance for upcoming XLA change (#3218)

d0k committed 5 years ago

Verified e3b046bc

Remove pe from name_stack and test. (#3209)

LenaMartens committed 5 years ago

Verified 1cc47192

remove stray print statement from #1529

mattjj committed 5 years ago

Verified 9f8a4ad3

Implement pmap of sharded_jit (#3144)

skye committed 5 years ago

Verified 6ffde806

Remove dtype warning for `np.quantile` (#3188)

fehiepsi committed 5 years ago

Verified e526109a

Add a JAX flag to avoid most optimizations. (#3208)

georgevdd committed 5 years ago

Verified 0f230029

Add a summary explaining the usage and context for JAX PRNG design. (#2525)

jblespiau committed 5 years ago

Verified a486f548

Fix error in code generation of batched while loops (#3207)

gnecula committed 5 years ago

Verified f18f7920

Fix experimental host callback on multi-host (#3200)

jekbradbury committed 5 years ago

Verified 0eace80a

Older