Commits

Breadcrumbs

History for

onmain

User selector

All users

Datepicker

All time

Commit History

Commits on Jun 17, 2024

• [ROCm] Bring-up pjrt support

  Ruturaj4

  committedJun 17, 2024
  99c2b7b

  Copy full SHA for 99c2b7b

Commits on Mar 25, 2024

• Remove support for CUDA 11.

  Show description for 0be07e6
  

  jax authors

  committedMar 25, 2024
  0be07e6

  Copy full SHA for 0be07e6

Commits on Mar 22, 2024

• Reverts bed4f65

  Show description for 910a31d
  

  jax authors

  committedMar 22, 2024
  910a31d

  Copy full SHA for 910a31d

• Remove support for CUDA 11.

  Show description for bed4f65
  

  jax authors

  committedMar 22, 2024
  bed4f65

  Copy full SHA for bed4f65

Commits on Jan 19, 2024

• [typing] fix incorrect annotation

  jakevdp

  committedJan 19, 2024
  3084c24

  Copy full SHA for 3084c24

Commits on Nov 6, 2023

• [PJRT C API] Change build wheel script to build a separate package for cuda kernels.

  Show description for 462ef16
  

  Jieying Luo

  authored and

  jax authors

  committedNov 6, 2023
  462ef16

  Copy full SHA for 462ef16

Commits on Sep 3, 2023

• Align the custom_call implementation in mlir and hlo_helpers.

  Show description for 24f9011
  

  gnecula

  authored and

  jax authors

  committedSep 3, 2023
  24f9011

  Copy full SHA for 24f9011

Commits on Aug 10, 2023

• Add batching rules to jax.lax.linalg.tridiagonal_solve.

  Show description for 7dfc8ff
  

  srvasude

  authored and

  jax authors

  committedAug 10, 2023
  7dfc8ff

  Copy full SHA for 7dfc8ff

Commits on Mar 4, 2023

• Internal change

  Show description for ad8c39a
  

  jax authors

  authored and

  hawkinsp

  committedMar 4, 2023
  ad8c39a

  Copy full SHA for ad8c39a

Commits on Feb 13, 2023

• fix change to csr lowering rule

  jakevdp

  committedFeb 13, 2023
  ddae1d0

  Copy full SHA for ddae1d0

Commits on Feb 11, 2023

• [sparse] implement BCSR.__matmul__

  jakevdp

  committedFeb 11, 2023
  de8a77a

  Copy full SHA for de8a77a

Commits on Dec 16, 2022

• (NFC) Prepare for migration from producing MHLO to producing StableHLO

  Show description for b8ae8e3
  

  Eugene Burmako

  authored and

  jax authors

  committedDec 16, 2022
  b8ae8e3

  Copy full SHA for b8ae8e3

Commits on Oct 25, 2022

• Merge CUDA and ROCM kernel code in jaxlib.

  Show description for a852710
  

  hawkinsp

  authored and

  jax authors

  committedOct 25, 2022
  a852710

  Copy full SHA for a852710

Commits on Oct 10, 2022

• Reapply: Use input-output aliasing for jaxlib GPU custom calls.

  Show description for 22cd505
  

  hawkinsp

  authored and

  jax authors

  committedOct 10, 2022
  22cd505

  Copy full SHA for 22cd505

Commits on Oct 7, 2022

• Revert: Use input-output aliasing for jaxlib GPU custom calls.

  Show description for 2693afa
  

  hawkinsp

  authored and

  jax authors

  committedOct 7, 2022
  2693afa

  Copy full SHA for 2693afa

• Use input-output aliasing for jaxlib GPU custom calls.

  Show description for 93b839a
  

  hawkinsp

  authored and

  jax authors

  committedOct 7, 2022
  93b839a

  Copy full SHA for 93b839a

Commits on Sep 22, 2022

• Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

  Show description for ba557d5
  

  hawkinsp

  authored and

  jax authors

  committedSep 22, 2022
  ba557d5

  Copy full SHA for ba557d5

Commits on Sep 12, 2022

• [sparse] Lower batch-mode bcoo_dot_genernal to cusparseSpMM.

  Show description for 3243e23
  

  tlu7

  authored and

  jax authors

  committedSep 12, 2022
  3243e23

  Copy full SHA for 3243e23

Commits on Aug 19, 2022

• [sparse] Add batch count and batch stride to matrix descriptors.

  Show description for d37b711
  

  tlu7

  authored and

  jax authors

  committedAug 19, 2022
  d37b711

  Copy full SHA for d37b711

Commits on Aug 18, 2022

• Internal change

  Show description for 840c966
  

  jax authors

  committedAug 18, 2022
  840c966

  Copy full SHA for 840c966

• Catch ModuleNotFoundError instead of ImportError.

  Show description for 1e241dc
  hawkinsp

  committedAug 18, 2022
  1e241dc

  Copy full SHA for 1e241dc

Commits on Jul 15, 2022

• [sparse] Update _validate_coo_mhlo in gpu_sparse.

  Show description for b421e24
  

  tlu7

  authored and

  jax authors

  committedJul 15, 2022
  b421e24

  Copy full SHA for b421e24

Commits on May 6, 2022

• Refactor custom call building code in jaxlib to use a helper function.

  Show description for 883cf2b
  

  hawkinsp

  authored and

  jax authors

  committedMay 6, 2022
  883cf2b

  Copy full SHA for 883cf2b

• Split CUDA and HIP C++ code in jaxlib into separate directories.

  Show description for 08c3c2e
  

  hawkinsp

  authored and

  jax authors

  committedMay 6, 2022
  08c3c2e

  Copy full SHA for 08c3c2e

• Merge remaining CUDA and ROCM Python code.

  Show description for 562e27d
  

  hawkinsp

  authored and

  jax authors

  committedMay 6, 2022
  562e27d

  Copy full SHA for 562e27d

Loading