ROCm/jax

Pull Requests Commits

make float and complex builtins error on Tracers

mattjj committed 5 years ago

7ab67756

Disable gradients of custom_linear_solve for complex dtypes (#2573)

shoyer committed 5 years ago

Verified 0958f270

Merge pull request #2654 from google/pfix

mattjj committed 5 years ago

Verified 91918432

fix jaxpr invar avals

mattjj committed 5 years ago

5f1f29e4

Fix up previous jaxpr.rst commit. (#2647)

skye committed 5 years ago

Verified f37f2351

Add type annotations to user-facing functions in lax.py (#2644)

hawkinsp committed 5 years ago

Verified 5174f6d5

Add a link to Google's codesearch for JAX. (#2646)

hawkinsp committed 5 years ago

Verified fcefc7e8

Merge pull request #2642 from jakevdp/trunc

mattjj committed 5 years ago

Verified e9910e0c

Merge pull request #2637 from jacobjinkelly/morejets

mattjj committed 5 years ago

Verified 52e779c3

Merge pull request #2631 from jakevdp/copysign

mattjj committed 5 years ago

Verified aabc3ac5

Add implementation of np.trunc

jakevdp committed 5 years ago

46a89229

Update scan jaxpr documentation. (#2641)

skye committed 5 years ago

Verified f8dc650b

Fully stage out scan under jit. (#2638)

skye committed 5 years ago

Verified f0ab673d

Merge branch 'master' of https://github.com/google/jax into morejets

jacobjinkelly committed 5 years ago

ab090e21

add expm1 and log1p

jacobjinkelly committed 5 years ago

4d7b63c5

Glob over subdirectories in top-level BUILD file. (#2636)

hawkinsp committed 5 years ago

Verified 2512ec6e

Set check_dtypes=False in copysign

jakevdp committed 5 years ago

49c6c229

Enable some GPU and TPU tests that now pass. (#2633)

hawkinsp committed 5 years ago

Verified f20a1c96

Implement np.copysign

jakevdp committed 5 years ago

bd00e591

Mark primitive parameters as keyword-only arguments in rules in lax.py. (#2625)

hawkinsp committed 5 years ago

Verified fa383b4a

Make median() more efficient & handle infs correctly (#2626)

jakevdp committed 5 years ago

Verified 5c74c396

jit: raise TypeError if called on generator function

jakevdp committed 5 years ago

f3a57656

Enable a disabled convolution test. (#2624)

hawkinsp committed 5 years ago

Verified 44e761b3

Add support for interpolation options in quantile() and percentile() (#2610)

jakevdp committed 5 years ago

Verified bbed6f8b

Remove unnecessary code in colabs (#2623)

Jin Dong committed 5 years ago

Verified 6213f8b8

Adjust pmap_bechmark.py values to be more realistic. (#2622)

skye committed 5 years ago

Verified 3fe8bd02

Add import from scipy.sparse (#2621)

shoyer committed 5 years ago

Verified e8f989e3

DOC: note how derivatives are computed for CG (#2619)

shoyer committed 5 years ago

Verified 1472eb3a

cumsum is linear, so its gradient can be linear also. (#2618)

hawkinsp committed 5 years ago

Verified cf4dd84b

Also enable nan checking for complex numbers (#2616)

ibab committed 5 years ago

Verified fc23e071

Older