Commits

Branch selector

 main

User selector

ezhulenev

Datepicker

All time

Commit History

Commits on May 24, 2024

• [xla:python] Add support for registering custom call targets for all XLA execution stages and for XLA FFI traits

  Show description for d5c7ccc
  

  ezhulenev

  authored and

  jax authors

  committed
  d5c7ccc

  Copy full SHA for d5c7ccc

Commits on May 22, 2024

• Move host-compute-offloading wrapping after SPMD sharding to shard calls executed on host

  Show description for 861999f
  

  ezhulenev

  authored and

  jax authors

  committed
  861999f

  Copy full SHA for 861999f

Commits on Mar 1, 2024

• [jax-triton] Do not capture jax-triton calls that require autotuning

  Show description for 1ae2022
  

  ezhulenev

  authored and

  jax authors

  committed
  1ae2022

  Copy full SHA for 1ae2022

Commits on Feb 23, 2024

• [jax-triton] Synchronize autotuning stream with a main one

  Show description for 3a69b80
  

  ezhulenev

  authored and

  jax authors

  committed
  3a69b80

  Copy full SHA for 3a69b80

Commits on Feb 2, 2024

• Disable JAX compilation cache for XLA:CPU

  Show description for 28ef77d
  

  ezhulenev

  authored and

  jax authors

  committed
  28ef77d

  Copy full SHA for 28ef77d

Commits on Jan 10, 2024

• [pjrt:cpu] Add CpuTopology to TfrtCpuClient and enable persistent compilation cache for cpu backend

  Show description for ba4c2b1
  

  ezhulenev

  authored and

  jax authors

  committed
  ba4c2b1

  Copy full SHA for ba4c2b1

Commits on Feb 2, 2023

• [jax] Skip compilation cache test for older jaxlibs

  Show description for 9d5132f
  

  ezhulenev

  authored and

  jax authors

  committed
  9d5132f

  Copy full SHA for 9d5132f

Commits on Sep 15, 2022

• [jax] Update expected error message in preparation for new XLA:GPU runtime

  Show description for 2fb8695
  

  ezhulenev

  authored and

  jax authors

  committed
  2fb8695

  Copy full SHA for 2fb8695