ROCm/jax

Pull Requests Commits

Remove noisy warnings for var() of int

jakevdp committed 5 years ago

7584b16d

Make dtype an explicit argument to nan-reductions

jakevdp committed 5 years ago

05cc7911

Add support for XLA variadic reduce

sharadmv committed 5 years ago

590e629f

Support shape parameter in jax.numpy.*_like()

jakevdp committed 5 years ago

ffaf1cdf

Merge pull request #4929 from jakevdp:out-kwds

a-googler committed 5 years ago

d70f2fb5

Merge pull request #4649 from qiuminxu:name_scope_jax2tf

a-googler committed 5 years ago

b8b3e5f0

qiuminxu committed 5 years ago

5a3d09de

Preserve name scope of named_call primitive in jax2tf

qiuminxu committed 5 years ago

769df737

Merge pull request #4796 from qiuminxu:add_jax_named_call

a-googler committed 5 years ago

69c920c6

Merge pull request #4927 from bchetioui:test_zeros_like

a-googler committed 5 years ago

06730476

Merge pull request #4921 from bchetioui:fix_argminmax_gpu

a-googler committed 5 years ago

09a0ed6e

Merge pull request #4936 from jakevdp:fix-mypy

a-googler committed 5 years ago

52d86899

Merge pull request #4912 from bchetioui:test_comparators

a-googler committed 5 years ago

3d067e82

Merge pull request #4934 from hawkinsp:bazel

a-googler committed 5 years ago

ac5d4c5c

jakevdp committed 5 years ago

9d3719f9

Update bazel version to 3.1.0.

hawkinsp committed 5 years ago

b222e5e0

Accept `None` as a trivial LazyExpr.

jblespiau committed 5 years ago

37a594ed

Merge pull request #4930 from hawkinsp:polymul

a-googler committed 5 years ago

8fafbf9b

Relax test tolerance on polymul tests.

hawkinsp committed 5 years ago

92cd0f52

lax_numpy: fix handling of keyword in wrapped methods

jakevdp committed 5 years ago

73d82f32

Merge pull request #4928 from jakevdp:in1d

a-googler committed 5 years ago

50571d48

Merge pull request #4883 from bchetioui:test_argmax_argmin

a-googler committed 5 years ago

6dc7f316

Update in1d() TODO

jakevdp committed 5 years ago

6fe2f776

[jax2tf] Add harness to test conversion of zeros_like.

bchetioui committed 5 years ago

9104125a

[jax2tf] Add testing for conversion of comparison operators.

bchetioui committed 5 years ago

356e38b1

[jax2tf] Added tests for the conversion of transpose.

bchetioui committed 5 years ago

e1e05140

Filter out non-integer integer_dtypes in argmin/max.

bchetioui committed 5 years ago

4d037996

Disable more tests

gnecula committed 5 years ago

06f60f46

Merge pull request #4901 from bchetioui:test_concatenate

a-googler committed 5 years ago

20e43046

[jax2tf] Add testing for argmax and argmin.

bchetioui committed 5 years ago

f9d78a2b

Newer Older