ROCm/jax

Pull Requests Commits

Adagrad optimizer

rohan-anil committed 6 years ago

a08f7ad5

Merge pull request #707 from hawkinsp/master

hawkinsp committed 6 years ago

Verified 6121ddfb

Merge pull request #708 from hawkinsp/long

hawkinsp committed 6 years ago

Verified c1dbeb0f

Incorporate review comments.

hawkinsp committed 6 years ago

696ef7ab

Add numpy.longlong to set of known constant types.

hawkinsp committed 6 years ago

6e9683f9

Add a test for long constant handling.

hawkinsp committed 6 years ago

62f4123b

Add support for 'long' in Python 2 mode to xla_bridge.py.

hawkinsp committed 6 years ago

b1cd2640

Fix copy-and-paste comment.

hawkinsp committed 6 years ago

29b1a30e

Add np.linalg.eig to the documentation.

hawkinsp committed 6 years ago

7e3295eb

Implement np.linalg.eig on CPU.

hawkinsp committed 6 years ago

a96944eb

Merge pull request #706 from hawkinsp/execute

hawkinsp committed 6 years ago

Verified 8c8314f7

Remove option to skip checks on Execute().

hawkinsp committed 6 years ago

38b3859c

Add lax.scan to docs.

hawkinsp committed 6 years ago

c2369ce3

fix typo unkown -> unknown

mattjj committed 6 years ago

dff8e60f

Merge pull request #704 from google/differentiable-scan

mattjj committed 6 years ago

Verified 29629931

mattjj committed 6 years ago

3c04dedc

add literals to jaxprs

mattjj committed 6 years ago

d27bc0a1

Merge pull request #703 from hawkinsp/master

hawkinsp committed 6 years ago

Verified 3ba51ef6

Revert to requiring Bazel 0.24 for JAX builds.

hawkinsp committed 6 years ago

5599f473

add pretty-printing to TypedJaxpr

mattjj committed 6 years ago

6e9718a2

improve core.typed_jaxpr arg typechecks

mattjj committed 6 years ago

65202821

remove unneeded packs from scan tests

mattjj committed 6 years ago

98ed8731

make tests pass with skip_checks = False

mattjj committed 6 years ago

4fcd96f9

scan bug fixed, other cleanup

mattjj committed 6 years ago

29e67f01

fix typo in pmap test

mattjj committed 6 years ago

78c80477

fix things we broke on the path to scan

mattjj committed 6 years ago

5cfa1801

Merge pull request #690 from google/collective-permute

mattjj committed 6 years ago

Verified 5c300f6c

sketch out ppermute transpose rule, untested

mattjj committed 6 years ago

134ce27f

Merge pull request #689 from google/improve-prng-shape-errors

mattjj committed 6 years ago

Verified 5be03cdf

Merge pull request #699 from hawkinsp/master

hawkinsp committed 6 years ago

Verified d05a5259

Newer Older