ROCm/jax

Pull Requests Commits

mattjj committed 5 years ago

4aa10cab

mattjj committed 5 years ago

b39942ce

an experiment in handling instances with __array__

mattjj committed 5 years ago

abc6599e

Raise an error on non-hashable static arguments for jax.jit and xla_computation.

jblespiau committed 5 years ago

cb48f423

tiny change to test source sync

froystig committed 5 years ago

a7de6942

Merge pull request #4597 from gnecula:tf_poly3

froystig committed 5 years ago

3e56be2a

Merge pull request #4716 from gnecula:update_changelog

a-googler committed 5 years ago

38ba3471

Update CHANGELOG

gnecula committed 5 years ago

da9a64ef

gnecula committed 5 years ago

8458cab0

Edits for the documentation

gnecula committed 5 years ago

36542e04

[jax2tf] Expanding jax2tf shape polymorphism.

gnecula committed 5 years ago

f0d2a4d2

Merge pull request #4713 from google:simpler-select-jvp

a-googler committed 5 years ago

9092ad3d

simplify select jvp

mattjj committed 5 years ago

7a73e99e

Merge pull request #4708 from google:docfix-jit

a-googler committed 5 years ago

ab3a4c2f

Merge pull request #4695 from Joshuaalbert:master

a-googler committed 5 years ago

a4956695

all changes plus test verifcation on TPU squashed

Joshuaalbert committed 5 years ago

231168d4

Merge pull request #4707 from google:traceback-register

a-googler committed 5 years ago

25b40702

Merge pull request #4709 from google:update-pypi

a-googler committed 5 years ago

4c9545d6

Merge pull request #4702 from fabianp:patch-3

a-googler committed 5 years ago

3bb0c49b

update version and changelog for pypi

mattjj committed 5 years ago

4ddfffff

fix jit docstring

froystig committed 5 years ago

9fc53b6e

move `wraps` from `api_util` to `util` to avoid cyclic dependencies

froystig committed 5 years ago

fa1d7ab5

add path exclusion opt-in to filtered stack traces and use it throughout the codebase

froystig committed 5 years ago

5d50e193

Merge pull request #4698 from google:custom-gradient-wrapper

a-googler committed 5 years ago

abbd8270

FIX: example usage of optimizers

fabianp committed 5 years ago

Verified 5d736286

add jax.custom_gradient wrapper for jax.custom_vjp

mattjj committed 5 years ago

3c6cdcfc

Merge pull request #4700 from hawkinsp:fft

a-googler committed 5 years ago

602c7836

Add PocketFFT mirror to fix download problems for some Bazel versions.

hawkinsp committed 5 years ago

9b7fd632

[JAX] Use PocketFFT for FFTs on CPU instead of Eigen.

hawkinsp committed 5 years ago

f58f1ee4

Merge pull request #4661 from jblespiau:changelist/338124676

a-googler committed 5 years ago

8121255d

Older