ROCm/jax

Pull Requests Commits

fix type error in jax2tf

mattjj committed 5 years ago

5c85baaa

remove lazy device constants, simplify LazyExpr

mattjj committed 5 years ago

4b2571f5

don't generate lazy iota/eye/tri/delta omnistaging

mattjj committed 5 years ago

4e65a6f0

Merge pull request #4532 from gnecula:tf_omnistaging

a-googler committed 5 years ago

a97cb870

[jax2tf] Ensure jax2tf still works without omnistaging

gnecula committed 5 years ago

9d9762b7

Merge pull request #4065 from skye:wheels

a-googler committed 5 years ago

0d69a84f

Use local version identifiers to distribute cuda jaxlib wheels.

skye committed 5 years ago

cacb0175

Merge pull request #4524 from jakevdp:fix-mean

a-googler committed 5 years ago

ae3d3101

Merge pull request #4526 from google:4510-2

a-googler committed 5 years ago

cd03bea5

mattjj committed 5 years ago

78fe038c

add test for scan input forwarding

mattjj committed 5 years ago

0aeeb63f

make scan fwd raw extensive inputs as DeviceArray

mattjj committed 5 years ago

5355776e

Merge pull request #4470 from gnecula:tf_tracers

a-googler committed 5 years ago

fc69d455

Fix extraneous dtype warning in jnp.mean

jakevdp committed 5 years ago

67a77c7e

[jax2tf] Port jax2tf to use omnistaging

gnecula committed 5 years ago

0213efdf

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

Older