ROCm/jax

Pull Requests Commits

remove mysterious code that is no longer needed

mattjj committed 5 years ago

55be9dd9

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

Merge pull request #4500 from LenaMartens:changelist/336082045

a-googler committed 5 years ago

684a5849

Recast int/bool tangents to float0 in custom_jvp/vjps

LenaMartens committed 5 years ago

e3d622cc

Merge pull request #4501 from gnecula:tf_tests_cf1

a-googler committed 5 years ago

95c0fea5

Enable TPU linalg tests that now pass.

hawkinsp committed 5 years ago

2fca9170

Add a flag to control the C++ jax.jit behavior.

jblespiau committed 5 years ago

6fec4974

[jax2tf] Ensure that in tests TF does not constant-fold in eager mode before compiling

gnecula committed 5 years ago

dc9168ba

Merge pull request #4475 from bchetioui:fix_static_tests

a-googler committed 5 years ago

8df116bd

Fix failing tests on TPU by avoiding using scalars

bchetioui committed 5 years ago

82995980

Merge pull request #4491 from google:remove-test

a-googler committed 5 years ago

83060cc1

Merge pull request #4489 from google:test-fixes

a-googler committed 5 years ago

e65115b8

remove an always-skipped test, redundant by now

froystig committed 5 years ago

30b8cc57

Merge pull request #4485 from hawkinsp:nan

a-googler committed 5 years ago

4b65f33f

remove stale version guard in test

froystig committed 5 years ago

1fb097bd

skip test that is broken at unreleased (source) jaxlib

froystig committed 5 years ago

16cd3304

Older