ROCm/jax

Pull Requests Commits

improve Parallel daxpr construct

mattjj committed 6 years ago

0fb88ee9

add print lines for last example

mattjj committed 6 years ago

2f171192

clean up daxpr language

mattjj committed 6 years ago

aaca3f9c

mattjj committed 6 years ago

1d2663f7

clarify that combinators don't return anything

mattjj committed 6 years ago

b1f90ee5

initial software pipeline experiments

mattjj committed 6 years ago

598eeb4e

Fix bazel version to 0.24.0 in Dockerfile to make sure we get a specific, working bazel version during builds.

hawkinsp committed 6 years ago

1d4de612

Set default Apple platform to fix TensorFlow build breakage when built for Jaxlib.

hawkinsp committed 6 years ago

f8eda9bb

Update XLA (TF) to unbreak build.

hawkinsp committed 6 years ago

0672b368

Merge pull request #534 from google/scipy-test-tolerances

mattjj committed 6 years ago

Verified 1d88e955

Increment jaxlib version in preparation for a new release.

hawkinsp committed 6 years ago

f3f4f3b1

Update XLA (TensorFlow) version to head.

hawkinsp committed 6 years ago

caf544a3

change scipy.special.log_ndtr test rng for numerics

mattjj committed 6 years ago

482c1ce8

Merge branch 'master' of github.com:google/jax

mattjj committed 6 years ago

0656dc22

fix pmap performance bug, dont always copy to host

mattjj committed 6 years ago

25479c35

Merge pull request #533 from hawkinsp/master

hawkinsp committed 6 years ago

Verified 2557de19

Fix minor issue in conj_p translation rule.

hawkinsp committed 6 years ago

fa6b06fd

Merge pull request #528 from kroq-gar78/cross

mattjj committed 6 years ago

Verified c9aa6010

Merge pull request #525 from google/jarrett-jvps-2

mattjj committed 6 years ago

Verified 18bc5936

Comment out test case (need to change tolerance)

kroq-gar78 committed 6 years ago

2b7d2df7

Merge pull request #530 from google/dot-dot-general

froystig committed 6 years ago

Verified f857762a

implement higher-rank cases of lax_numpy.dot in terms of lax.dot_general, avoiding a reshape

froystig committed 6 years ago

323353eb

Fix dtypes in cross product

kroq-gar78 committed 6 years ago

0a5a633d

Implement cross product and test cases

kroq-gar78 committed 6 years ago

48934dc9

fix typo in linearize docstring

mattjj committed 6 years ago

850e8a75

attempt to fix readtheddocs.io formatting

mattjj committed 6 years ago

5704624c

add jax.linearize to jax.readthedocs.io

mattjj committed 6 years ago

f5b4391f

Merge pull request #527 from google/issue526

mattjj committed 6 years ago

Verified a9ab470e

mattjj committed 6 years ago

595a9800

add clarification about linearize vs jvp+vmap

mattjj committed 6 years ago

54ac8795

Older