Commits

Breadcrumbs

History for

onmaster

User selector

All users

Datepicker

All time

Commit History

Commits on Mar 25, 2024

• Remove support for CUDA 11.

  Show description for 0be07e6
  

  jax authors

  committed
  0be07e6

Commits on Mar 22, 2024

• Reverts bed4f65

  Show description for 910a31d
  

  jax authors

  committed
  910a31d

• Remove support for CUDA 11.

  Show description for bed4f65
  

  jax authors

  committed
  bed4f65

Commits on Nov 17, 2023

• Use MLIR generated convenience functions athing(...) instead of writing AThingOp(...).result.

  Show description for 8e8dc26
  

  hawkinsp

  authored and

  jax authors

  committed
  8e8dc26

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

  committed
  462ef16

Commits on Sep 19, 2023

• Run LSTM test using FP32 math (as opposed to TF32)

  Show description for fc1c31d
  andportnoy

  committed
  fc1c31d

Commits on Sep 11, 2023

• Improve the gpu lowering error message if users forget link the gpu library.

  Show description for 997b35e
  

  maxwillzq

  authored and

  jax authors

  committed
  997b35e

Commits on Jun 6, 2023

• Resolve unsoundness caught by pytype --strict-none-binding

  Show description for 9c2eba3
  

  jax authors

  committed
  9c2eba3

Commits on May 3, 2023

• update cuDNN RNN code not to save 'workspace' scratch between fwd and bwd

  Show description for 56feaca
  

  mattjj

  authored and

  jax authors

  committed
  56feaca

Commits on May 1, 2023

• Remove tupling of custom call results.

  Show description for ba11b9d
  

  hawkinsp

  authored and

  jax authors

  committed
  ba11b9d

Commits on Mar 4, 2023

• Internal change

  Show description for ad8c39a
  

  jax authors

  authored and

  hawkinsp

  committed
  ad8c39a

Commits on Dec 27, 2022

• Migrate JAX from producing MHLO to producing StableHLO

  Show description for a1480c4
  

  Eugene Burmako

  authored and

  jax authors

  committed
  a1480c4

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

  committed
  b8ae8e3

Commits on Nov 28, 2022

• Add CUDNN custom call for LSTM. Exposed as jax.experimental.rnn module.

  Show description for 4d1c4bc
  

  zhangqiaorjc

  authored and

  jax authors

  committed
  4d1c4bc

Commits on Nov 23, 2022

• Rollback of "Add CUDNN custom call for LSTM. Exposed as jax.experimental.rnn module."

  Show description for d1fbdbc
  

  jax authors

  committed
  d1fbdbc

• Add CUDNN custom call for LSTM. Exposed as jax.experimental.rnn module.

  Show description for 78963b6
  

  zhangqiaorjc

  authored and

  jax authors

  committed
  78963b6

Loading