ROCm/jax

Pull Requests Commits

Enable the C++ jax.jit fast code-path by default.

jblespiau committed 5 years ago

90b81b57

Merge pull request #4518 from gnecula:tf_debug

a-googler committed 5 years ago

e194dff6

[jax2tf] Fix the behavior for shift when the shift amount is negative or large.

gnecula committed 5 years ago

a1e67be7

Merge pull request #4517 from google:issue4510

a-googler committed 5 years ago

e6399cd7

make pytype happy

mattjj committed 5 years ago

7691be92

optimize scan partial_eval to fix #4510

mattjj committed 5 years ago

52fe026c

Merge pull request #4514 from google:remove-unneeded-code

a-googler committed 5 years ago

d4da9cc1

Merge pull request #4515 from google:enable-conv-test

a-googler committed 5 years ago

ab462baf

Merge pull request #4512 from jakevdp:raise-to-shaped

a-googler committed 5 years ago

4994b82b

re-enable lax autodiff test

froystig committed 5 years ago

1614e283

remove mysterious code that is no longer needed

mattjj committed 5 years ago

55be9dd9

Merge pull request #4511 from google:debug-nans-test-move

a-googler committed 5 years ago

91492ca1

lax_control_flow: retry via function rather than via loop

jakevdp committed 5 years ago

ef25da84

Fallback to Python for float0 in the C++ jax.jit.

jblespiau committed 5 years ago

008eae66

move a debug_nans test into debug_nans test file

mattjj committed 5 years ago

24de811a

Merge pull request #4161 from jakevdp:raise-to-shaped

a-googler committed 5 years ago

ddcfd458

Merge pull request #4483 from google:debug-nans-no-store-error

a-googler committed 5 years ago

3be8188c

Merge pull request #4483 from google:debug-nans-no-store-error

a-googler committed 5 years ago

57ec9dd9

wait for result in debug_nans_test

mattjj committed 5 years ago

09f2be15

Merge pull request #4508 from zhangqiaorjc:expm_test_slow

a-googler committed 5 years ago

2b76bcca

Merge branch 'master' into debug-nans-no-store-error

mattjj committed 5 years ago

4dd802c5

Skip 2nd order grad in expm tests.

zhangqiaorjc committed 5 years ago

afef644d

Merge pull request #4507 from hawkinsp:xla

a-googler committed 5 years ago

a2a6dca4

fix weak_type issues in while_loop

jakevdp committed 5 years ago

a20b00a3

hawkinsp committed 5 years ago

72be97c8

fix weak_type issues in while_loop

jakevdp committed 5 years ago

0f0fa53b

fix weak type issues in scan

jakevdp committed 5 years ago

3ca6e5a1

raise_to_shaped: preserve weak_type by default

jakevdp committed 5 years ago

63933497

Merge pull request #4498 from jblespiau:changelist/333713171

a-googler committed 5 years ago

ab48a9af

Merge pull request #4502 from hawkinsp:linalg

a-googler committed 5 years ago

145ac406

Older