ROCm/jax

Pull Requests Commits

[JAX] fix jax and wheelhouse relative paths

Ruturaj4 committed 2 years ago

a92f7f05

[ROCm] Fix invalid repo url for EL path

mrodden committed 2 years ago

de61e3fc

Fix formatting issues

mrodden committed 2 years ago

96a25861

Fix DOCKER_BUILDKIT issue for jax CI

kiran-thumma committed 2 years ago

70ee4c23

Add docker builds for ubu22 and 24

mrodden committed 2 years ago

a2ab4acf

ayaka14732 committed 2 years ago

f02f2a7c

[ROCm] add build file.

Ruturaj4 committed 2 years ago

9b9ea337

* Add conditional docker interactive mode

mrodden committed 2 years ago

a26aa082

[ROCm] improve gpu script

Ruturaj4 committed 2 years ago

3192b8ac

[ROCm] Fix python in rocm ci_build script.

Ruturaj4 committed 2 years ago

1d45b98e

Add missing CPython build deps for pyenv

mrodden committed 2 years ago

f0a6a8b3

Fix not finding wheels in bazel output

mrodden committed 2 years ago

74401c33

Fix jenkins notty issue

mrodden committed 2 years ago

f4e1188b

Add defaults for ROCm build vars

mrodden committed 2 years ago

d7481d2d

Add copyright and license headers to new files

mrodden committed 2 years ago

04373f20

Add support for ROCm development builds

mrodden committed 2 years ago

b92f5ec9

Add internal release support to get_rocm.py

mrodden committed 2 years ago

f8362e38

[ROCm] Change ROCm builds to manylinux wheels

mrodden committed 2 years ago

3a1601dd

Fix lint in run_single_gpu.py

yueshengys committed 2 years ago

91079db3

[ROCM] add pytest reportlog

Ruturaj4 committed 2 years ago

534b7a33

[JAX] Fix run_single_gpu.py to include correct paths

Ruturaj4 committed 2 years ago

95ae95ca

[ROCm]: Add support to continue on fail

Rahul Batra committed 2 years ago

84a3e91c

Start JAX and jaxlib 0.4.31 release

yashk2810 committed 2 years ago

7fd93027

Merge pull request #22703 from Rifur13:plugin-fix

a-googler committed 2 years ago

f070c065

Merge pull request #19760 from Blair-Johnson:fix-pytree-grads-sparse

a-googler committed 2 years ago

9beb4f14

Re-pack gradients of jax.experimental.sparse.grad() to match original pytrees & test cases

Blair-Johnson committed 2 years ago

802a14cd

Merge pull request #22690 from jakevdp:inplace-doc

a-googler committed 2 years ago

85e83b50

Bump the minimum CUDNN version to v9.1.

hawkinsp committed 2 years ago

d1c0d993

Ignore the Deprecation warning produced about `native_serialization=False`.

belitskiy committed 2 years ago

6127baa1

Bump minimum SciPy version to 1.10.

hawkinsp committed 2 years ago

fd23b873

Older