ROCm/jax

Pull Requests Commits

Fix integer overflow in np.split on windows.

hawkinsp committed 5 years ago

a07c5e11

Support 'initial' argument in jax.numpy reductions

jakevdp committed 5 years ago

841ea14c

Merge pull request #4970 from jakevdp:reduction-tests

a-googler committed 5 years ago

a96aca5e

Add support for multi-host partitioning when using pmap(sharded_jit).

skye committed 5 years ago

4e56cf96

Merge pull request #4988 from hawkinsp:windows

a-googler committed 5 years ago

7d135648

Fix lax_reference population_count implementation to avoid overflowing Python int -> np.uint32 casts.

hawkinsp committed 5 years ago

f8ce7d88

Merge pull request #4986 from hawkinsp:windows

a-googler committed 5 years ago

ef5218f6

Merge pull request #4985 from jakevdp:cleanup-todo

a-googler committed 5 years ago

882b59e0

Add np.intc to the set of valid jaxtypes.

hawkinsp committed 5 years ago

d2a0bbd9

Increase test coverage for reductions.

jakevdp committed 5 years ago

7140431a

Cleanup: remove obsolete TODOs

jakevdp committed 5 years ago

4e738bf5

[jax2tf] Fix and test conversion of iota

bchetioui committed 5 years ago

e4d5be27

[jax2tf] Added testing for real_p/imag_p

bchetioui committed 5 years ago

e65ed8b6

[jax2tf] Add testing for the conversion of complex_p.

bchetioui committed 5 years ago

559e5077

[jax2tf] Added tests for the conversion of reducers.

bchetioui committed 5 years ago

f7480d65

Merge pull request #4963 from bchetioui:test_div

a-googler committed 5 years ago

8b6be205

Change jaxlib build rules to build a wheel, rather than writing output to the source directory.

hawkinsp committed 5 years ago

c06ead6b

Add derivatives for eigenvalues (not eigenvectors)

j-towns committed 5 years ago

5fccc89a

Merge pull request #4953 from apaszke:remove-axis-env-devices

a-googler committed 5 years ago

5a41779f

Reenable pytype for previously problematic ABSL library.

hawkinsp committed 5 years ago

e4e1af5d

Merge pull request #4974 from hawkinsp:flags

a-googler committed 5 years ago

b16621b1

Add -Wno-stringop-truncation to build flags on Linux.

hawkinsp committed 5 years ago

1fcbd2f0

Merge pull request #4969 from jakevdp:fix-reductions

a-googler committed 5 years ago

1e715121

Internal change

hawkinsp committed 5 years ago

c8e6a6a0

jax.numpy: match numpy arguments for reductions

jakevdp committed 5 years ago

cc53ae04

Merge pull request #4949 from hawkinsp:abstractarrays

a-googler committed 5 years ago

541e8ab8

Merge pull request #4965 from hawkinsp:build

a-googler committed 5 years ago

9aa04443

Short-circuit references to jax.core via jax.abstract_arrays.

hawkinsp committed 5 years ago

424594fe

Merge pull request #4938 from jakevdp:nan-red-dtype

a-googler committed 5 years ago

28038880

Merge pull request #4867 from jakevdp:indexing-typeerror

a-googler committed 5 years ago

9f631897

Newer Older