Commits

Breadcrumbs

History for

onmain

User selector

All users

Datepicker

All time

Commit History

Commits on Aug 23, 2024

• Refactor hermetic CUDA flags and update --config=cuda to add CUDA dependencies both for bazel build and bazel test phases.

  Show description for 279977c
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  279977c

  Copy full SHA for 279977c

Commits on Aug 14, 2024

• Introduce hermetic CUDA in Google ML projects.

  Show description for 599c13a
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  599c13a

  Copy full SHA for 599c13a

Commits on Aug 13, 2024

• Disable TensorRT in TF, XLA and JAX.

  Show description for 9f68576
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  9f68576

  Copy full SHA for 9f68576

Commits on Aug 6, 2024

• Remove enable_gpu and xla_python_enable_gpu from jax .bazelrc.

  Show description for abe7982
  

  Jieying Luo

  authored and

  jax authors

  committed
  abe7982

  Copy full SHA for abe7982

Commits on Aug 1, 2024

• Rollback as it broke some tests.

  Show description for bc0229a
  

  Jieying Luo

  authored and

  jax authors

  committed
  bc0229a

  Copy full SHA for bc0229a

• Cleanup. Remove build:cuda_plugin and set enable_gpu and xla_python_enable_gpu to false in build:cuda.

  Show description for ff17b76
  

  Jieying Luo

  authored and

  jax authors

  committed
  ff17b76

  Copy full SHA for ff17b76

Commits on Jul 10, 2024

• Upgrade to support and default to clang 18 for the OSS compiler

  Show description for cddb8f2
  

  MichaelHudgins

  authored and

  jax authors

  committed
  cddb8f2

  Copy full SHA for cddb8f2

Commits on Jun 26, 2024

• Update minimum Python version to 3.10.

  hawkinsp

  committed
  945fde4

  Copy full SHA for 945fde4

Commits on Jun 17, 2024

• [ROCm] Bring-up pjrt support

  Ruturaj4

  committed
  99c2b7b

  Copy full SHA for 99c2b7b

Commits on May 30, 2024

• [Mosaic GPU] Integrate the ExecutionEngine with the jaxlib GPU plugin

  Show description for cfe64cd
  

  apaszke

  authored and

  jax authors

  committed
  cfe64cd

  Copy full SHA for cfe64cd

Commits on May 24, 2024

• Update cuDNN to version 9.1.1 in JAX

  Show description for 15a1985
  

  beckerhe

  authored and

  jax authors

  committed
  15a1985

  Copy full SHA for 15a1985

Commits on May 22, 2024

• Remove redundant commands from the RBE cross-compile build scripts

  Show description for 0f4dff5
  

  nitins17

  authored and

  jax authors

  committed
  0f4dff5

  Copy full SHA for 0f4dff5

Commits on May 16, 2024

• Use hermetic Python in JAX, see "Managing hermetic Python" in developer.md for details

  Show description for 517e299
  

  vam-google

  authored and

  jax authors

  committed
  517e299

  Copy full SHA for 517e299

Commits on May 6, 2024

• Add a config for using Clang on Windows.

  Show description for fb65ba4
  

  jax authors

  committed
  fb65ba4

  Copy full SHA for fb65ba4

Commits on Apr 18, 2024

• Initial commit for Mosaic GPU

  Show description for 8e3f5b1
  

  apaszke

  authored and

  jax authors

  committed
  8e3f5b1

  Copy full SHA for 8e3f5b1

Commits on Mar 25, 2024

• Remove support for CUDA 11.

  Show description for 0be07e6
  

  jax authors

  committed
  0be07e6

  Copy full SHA for 0be07e6

Commits on Mar 22, 2024

• Reverts bed4f65

  Show description for 910a31d
  

  jax authors

  committed
  910a31d

  Copy full SHA for 910a31d

• Remove support for CUDA 11.

  Show description for bed4f65
  

  jax authors

  committed
  bed4f65

  Copy full SHA for bed4f65

Commits on Mar 20, 2024

• Add an experimental build-only continuous cross compile build for MacOS x86

  Show description for 3f13308
  

  kanglant

  authored and

  jax authors

  committed
  3f13308

  Copy full SHA for 3f13308

Commits on Mar 7, 2024

• Add an experimental build-only continuous cross compile build for Linux Aarch64

  Show description for f0afc1b
  

  kanglant

  authored and

  jax authors

  committed
  f0afc1b

  Copy full SHA for f0afc1b

Commits on Feb 3, 2024

• Add --use_clang and --clang_path options to build.py

  Show description for 6928465
  

  ddunl

  authored and

  jax authors

  committed
  6928465

  Copy full SHA for 6928465

Commits on Jan 10, 2024

• Enable Bazel remote cache in macOS continuous builds

  Show description for b58772c
  

  nitins17

  authored and

  jax authors

  committed
  b58772c

  Copy full SHA for b58772c

Commits on Jan 8, 2024

• Update CUDA to 12.3 in JAX/TF/XLA CIs

  Show description for 3d608b1
  

  jax authors

  committed
  3d608b1

  Copy full SHA for 3d608b1

Commits on Dec 5, 2023

• Change bazel config name "rbe_cpu_linux_py39" to "rbe_cpu_linux_py3.9"to be consistent with cuda bazel build configs.

  Show description for f25a51e
  

  Jieying Luo

  authored and

  jax authors

  committed
  f25a51e

  Copy full SHA for f25a51e

Commits on Nov 28, 2023

• Build CUDA kernels for sm_50 instead of sm_52.

  Show description for 6fa506c
  

  hawkinsp

  authored and

  jax authors

  committed
  6fa506c

  Copy full SHA for 6fa506c

Commits on Nov 20, 2023

• Reverts 2aaa755

  Show description for dce6ab4
  

  jax authors

  committed
  dce6ab4

  Copy full SHA for dce6ab4

Commits on Nov 10, 2023

• Reverts a80d35f

  Show description for 2aaa755
  

  jax authors

  committed
  2aaa755

  Copy full SHA for 2aaa755

• Replace gcc with clang compiler in CI nightly jobs.

  Show description for a80d35f
  

  jax authors

  committed
  a80d35f

  Copy full SHA for a80d35f

Commits on Nov 8, 2023

• Register plugin profiler for TPU and remove --config=tpu/--enable_tpu in jaxlib.

  Show description for 0ce7c7b
  

  Jieying Luo

  authored and

  jax authors

  committed
  0ce7c7b

  Copy full SHA for 0ce7c7b

Commits on Nov 3, 2023

• Replace gcc with clang compiler in presubmit and postsubmit CI Kokoro jobs.

  Show description for 59192b0
  

  jax authors

  committed
  59192b0

  Copy full SHA for 59192b0

Commits on Oct 24, 2023

• Reverts 9b1a656

  Show description for 8d49f9a
  

  hawkinsp

  authored and

  jax authors

  committed
  8d49f9a

  Copy full SHA for 8d49f9a

Commits on Oct 23, 2023

• Reverts 84c5169

  Show description for 9b1a656
  

  jax authors

  committed
  9b1a656

  Copy full SHA for 9b1a656

Commits on Oct 16, 2023

• Build jaxlib without PJRT GPU deps when plugin will be built.

  Show description for 0290150
  

  Jieying Luo

  authored and

  jax authors

  committed
  0290150

  Copy full SHA for 0290150

Commits on Oct 13, 2023

• Delete cuda 12.0.1 rbe configs since JAX doesn't support it anymore

  Show description for f5a1439
  

  yashk2810

  authored and

  jax authors

  committed
  f5a1439

  Copy full SHA for f5a1439

Commits on Oct 12, 2023

• Set RPATH, not RUNPATH in JAX CUDA builds.

  Show description for d856ecc
  hawkinsp

  committed
  d856ecc

  Copy full SHA for d856ecc

Pagination

Previous

Next