ROCm/jax

Pull Requests Commits

add remat docstring (#3542)

mattjj committed 6 years ago

Verified 696958d2

Fix typos pytrees page on readthedocs (#3548)

IgorWilbert committed 6 years ago

Verified 7e5407c7

Fix failure in PyTorch array interoperability test in non-x64 mode. (#3546)

hawkinsp committed 6 years ago

Verified a9c7413f

Remove warning suppression for tuple and list arguments to reductions. (#3545)

hawkinsp committed 6 years ago

Verified e680304d

Clarify docstrings regarding usage of static arguments in jit and vmap. (#3484)

JonyEpsilon committed 6 years ago

Verified 677baa54

Avoid direct type/dtype comparisons to fix NumPy 1.19 deprecation war… (#3543)

hawkinsp committed 6 years ago

Verified f036f5dd

Fix test failures due to RuntimeWarning in variance tests. (#3541)

hawkinsp committed 6 years ago

Verified 90894a28

Restrict .compress tests to arrays only. (#3540)

hawkinsp committed 6 years ago

Verified 28262da1

Merge quantile and nanquantile implementations. (#3539)

hawkinsp committed 6 years ago

Verified a422f635

Add np.diag_indices_from (#3500)

ashridh committed 6 years ago

Verified a44bc0c2

Add missing functions to documentation. (#3538)

hawkinsp committed 6 years ago

Verified fede6d90

Implement nan-quantile/median/percentile (#2397)

yurodiviy committed 6 years ago

Verified c357aadc

Fixed issue with the 'compress' method of 'jnp.ndarray' (#3537)

nova77 committed 6 years ago

Verified db2291bd

Future warning about lists and tuples (#3369)

joaogui1 committed 6 years ago

Verified 319eeaf5

Add np.unwrap (#3527)

clemisch committed 6 years ago

Verified a6e3f992

de-duplicate constants when lowering to XLA

froystig committed 6 years ago

edcd43ac

Support `b` and `return_sign` in scipy.special.logsumexp (#3488)

dpfau committed 6 years ago

Verified 9d173c62

avoid value-based error check in random.choice (#3531)

mattjj committed 6 years ago

Verified 74ee2ef6

add back a full_lower, dropped in #3491 (#3530)

mattjj committed 6 years ago

Verified a45e2837

refactor call primitives, simpler param processing (#3491)

mattjj committed 6 years ago

Verified 75278309

lax.sort: allow any sequence of Arrays, not just tuples (#3367)

jakevdp committed 6 years ago

Verified d5a5d301

Fix lax_reference's round for edge case inputs (#3309)

majnemer committed 6 years ago

Verified 7cc26505

Fix memory leak when no axis is provided to pmap. (#3394)

hawkinsp committed 6 years ago

Verified 86fcfbfa

Adds boolean support for bitwise not and unittests for boolean support on logical operations. (#3483)

thomaskeck committed 6 years ago

Verified 490c8533

Remove unnecessary static_argnum in np.gradient (#3512)

clemisch committed 6 years ago

Verified 5ee6bc00

fix an issue with newer versions of pytype (#3526)

mattjj committed 6 years ago

Verified 02494924

Add jax.scipy.signal.detrend (#3516)

jakevdp committed 6 years ago

Verified 33c455a1

Cast int8 to bool for lax.not in jax2tf. (#3519)

tomhennigan committed 6 years ago

Verified ca5b0b18

Fix typo: np.bool -> np.bool_ (#3525)

NeilGirdhar committed 6 years ago

Verified 046006e0

remove the lower_fun default multiple_results=True (#3524)

mattjj committed 6 years ago

Verified 2f7108f7

Newer Older