Commits

Branch selector

 main

User selector

hawkinsp

Datepicker

All time

Commit History

Commits on Sep 9, 2022

• Remove jax._src from JAX namespace.

  Show description for 40c80d7
  

  hawkinsp

  authored and

  jax authors

  committed
  40c80d7

  Copy full SHA for 40c80d7

Commits on Sep 8, 2022

• Drop support for non-MLIR modules in compile().

  Show description for 8b41888
  hawkinsp

  committed
  8b41888

  Copy full SHA for 8b41888

• Bump the minimum jaxlib version to 0.3.15.

  hawkinsp

  committed
  6c59d72

  Copy full SHA for 6c59d72

• Reapply: Use MLIR bytecode when passing IR to backends.

  Show description for 6ddf3c4
  

  hawkinsp

  authored and

  jax authors

  committed
  6ddf3c4

  Copy full SHA for 6ddf3c4

Commits on Sep 7, 2022

• Revert: Use MLIR bytecode when passing IR to backends.

  Show description for ccdcab2
  

  hawkinsp

  authored and

  jax authors

  committed
  ccdcab2

  Copy full SHA for ccdcab2

• Roll forward: Upgrade logistic into a primitive.

  Show description for 57b5acf
  

  hawkinsp

  authored and

  jax authors

  committed
  57b5acf

  Copy full SHA for 57b5acf

• Use MLIR bytecode when passing IR to backends.

  Show description for f62ab8f
  

  hawkinsp

  authored and

  jax authors

  committed
  f62ab8f

  Copy full SHA for f62ab8f

Commits on Sep 6, 2022

• Avoid direct references to jax._src.

  Show description for 9c79439
  

  hawkinsp

  authored and

  jax authors

  committed
  9c79439

  Copy full SHA for 9c79439

Commits on Aug 26, 2022

• [JAX] Handle non-contiguous GPU IDs in NCCL collectives.

  Show description for b9d7e05
  

  hawkinsp

  authored and

  jax authors

  committed
  b9d7e05

  Copy full SHA for b9d7e05

• Add missing header to ducc build file.

  hawkinsp

  committed
  20475cc

  Copy full SHA for 20475cc

• Upgrade logistic (sigmoid) function into a lax primitive.

  Show description for f68f1c0
  

  hawkinsp

  authored and

  jax authors

  committed
  f68f1c0

  Copy full SHA for f68f1c0

• Bump XLA version.

  hawkinsp

  committed
  12cb55d

  Copy full SHA for 12cb55d

• Fixes for PocketFFT->ducc migration.

  Show description for b63801b
  hawkinsp

  committed
  b63801b

  Copy full SHA for b63801b

• Update IREE backend name from "dylib" to "llvm-cpu".

  hawkinsp

  committed
  813ee0d

  Copy full SHA for 813ee0d

Commits on Aug 25, 2022

• [JAX] Deprecate .to_py() property on arrays. Implement __array__ instead.

  Show description for 5527966
  

  hawkinsp

  authored and

  jax authors

  committed
  5527966

  Copy full SHA for 5527966

Commits on Aug 24, 2022

• Upgrade logistic (sigmoid) function into a lax primitive.

  Show description for 6276194
  

  hawkinsp

  authored and

  jax authors

  committed
  6276194

  Copy full SHA for 6276194

• Avoid more direct references to jax._src without imports.

  Show description for 78cb9f8
  

  hawkinsp

  authored and

  jax authors

  committed
  78cb9f8

  Copy full SHA for 78cb9f8

Commits on Aug 23, 2022

• Suppress msan failure in PRNG code.

  Show description for 160a6c5
  

  hawkinsp

  authored and

  jax authors

  committed
  160a6c5

  Copy full SHA for 160a6c5

• [JAX] Prepare not to export jax._src by default.

  Show description for 335b2cf
  

  hawkinsp

  authored and

  jax authors

  committed
  335b2cf

  Copy full SHA for 335b2cf

Commits on Aug 22, 2022

• Increase some test shardings to reduce CI timeouts under asan/tsan/msan.

  Show description for da4e79a
  

  hawkinsp

  authored and

  jax authors

  committed
  da4e79a

  Copy full SHA for da4e79a

• Relax test tolerance to fix GPU CI failure.

  hawkinsp

  committed
  c247b9b

  Copy full SHA for c247b9b

Commits on Aug 19, 2022

• Disabled dlpack bfloat16 test for JAX<->cupy.

  Show description for 439ba1d
  hawkinsp

  committed
  439ba1d

  Copy full SHA for 439ba1d

Commits on Aug 18, 2022

• Be more selective about which MLIR pieces we build.

  Show description for 0839958
  hawkinsp

  committed
  0839958

  Copy full SHA for 0839958

• Update XLA commit.

  hawkinsp

  committed
  6128cb7

  Copy full SHA for 6128cb7

• Fix jax_cuda_visible_devices for distributed GPU clusters.

  Show description for accd189
  

  hawkinsp

  authored and

  jax authors

  committed
  accd189

  Copy full SHA for accd189

• Catch ModuleNotFoundError instead of ImportError.

  Show description for 1e241dc
  hawkinsp

  committed
  1e241dc

  Copy full SHA for 1e241dc

• Revert: Sparse direct solver using QR factorization from cuSOLVER. This is the jaxlib implementation. We will want to combine this with the sparse libraries already existing in JAX.

  Show description for 3bb0030
  

  hawkinsp

  authored and

  jax authors

  committed
  3bb0030

  Copy full SHA for 3bb0030

Commits on Aug 17, 2022

• Include ABI tag in jaxlib wheels.

  Show description for 5b0686f
  hawkinsp

  committed
  5b0686f

  Copy full SHA for 5b0686f

• Update XLA commit.

  Show description for 87db8fc
  hawkinsp

  committed
  87db8fc

  Copy full SHA for 87db8fc

Commits on Aug 16, 2022

• Bump test tolerance for eigenvalue test.

  Show description for 1e242be
  hawkinsp

  committed
  1e242be

  Copy full SHA for 1e242be

• Recommend using the cuda-nvcc package from the "nvidia" conda channel to fetch ptxas.

  hawkinsp

  committed
  8f1a346

  Copy full SHA for 8f1a346

• build.py fixes.

  Show description for 03876bd
  hawkinsp

  committed
  03876bd

  Copy full SHA for 03876bd

Commits on Aug 15, 2022

• Bump XLA version.

  Show description for 5603cb3
  hawkinsp

  committed
  5603cb3

  Copy full SHA for 5603cb3

• Point to the conda-forge jaxlib wheels in the JAX readme.

  hawkinsp

  committed
  026e760

  Copy full SHA for 026e760

Commits on Aug 14, 2022

• Import .bazelrc.user to allow users to override bazel options easily.

  hawkinsp

  committed
  184f8ab

  Copy full SHA for 184f8ab

Pagination

Previous

Next