ROCm/jax

Pull Requests Commits

typo fixes in pmap docstring

mattjj committed 6 years ago

52a2fb32

add pmap to sphinx docs

mattjj committed 6 years ago

d1994716

add pmap docstring

mattjj committed 6 years ago

38c6a8e8

Reenable some indexing tests that now pass.

hawkinsp committed 6 years ago

d9dc23a2

Update XLA build.

hawkinsp committed 6 years ago

bdb54e28

Merge pull request #713 from hawkinsp/master

hawkinsp committed 6 years ago

Verified eef97b6f

Expose jax.tree_util in the JAX docs.

hawkinsp committed 6 years ago

8c2c1a2e

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

Older