Commits

Branch selector

 main

User selector

ayaka14732

Datepicker

All time

Commit History

Commits on Sep 19, 2024

• [Pallas TPU] Add lowering for 64 bit

  ayaka14732

  committed
  de23fdb

  Copy full SHA for de23fdb

• Enable Pallas ops_test on GPU in 64-bit mode.

  Show description for 3f23866
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  3f23866

  Copy full SHA for 3f23866

Commits on Sep 10, 2024

• [Pallas] Simplify lowering and fix the test for lax.erf_inv_p

  Show description for 46bcb1e
  

  ayaka14732

  authored and

  jax authors

  committed
  46bcb1e

  Copy full SHA for 46bcb1e

• [Pallas] Fix array indexing error when dimension size is not a multiple of stride

  ayaka14732

  committed
  e0faa59

  Copy full SHA for e0faa59

Commits on Sep 9, 2024

• [Pallas GPU] Fix the behavior of jnp.sign(jnp.nan) and move the TPU test case for jnp.sign into the general test

  Show description for 7d2f0a7
  

  ayaka14732

  authored and

  jax authors

  committed
  7d2f0a7

  Copy full SHA for 7d2f0a7

Commits on Aug 29, 2024

• [Pallas GPU] Add 32-bit lowering rule for lax.erf_inv

  Show description for 1dff3a2
  

  ayaka14732

  authored and

  jax authors

  committed
  1dff3a2

  Copy full SHA for 1dff3a2

Commits on Aug 28, 2024

• Disable implicit type conversion during type matching

  ayaka14732

  committed
  f479350

  Copy full SHA for f479350

• Support strided load / store in interpret mode

  Show description for fc1af8d
  

  ayaka14732

  authored and

  jax authors

  committed
  fc1af8d

  Copy full SHA for fc1af8d

Commits on Aug 27, 2024

• Fix mypy error

  ayaka14732

  committed
  859eacb

  Copy full SHA for 859eacb

Commits on Aug 19, 2024

• Normalize "interpreter mode" to "interpret mode", and "InterpreterTest" to "InterpretTest"

  Show description for 36739e8
  

  ayaka14732

  authored and

  jax authors

  committed
  36739e8

  Copy full SHA for 36739e8

Commits on Jul 26, 2024

• Move TPU ops test to ops_test.py

  Show description for bb160cf
  

  ayaka14732

  authored and

  jax authors

  committed
  bb160cf

  Copy full SHA for bb160cf

• Add lowering for lax.sign

  ayaka14732

  committed
  6cc0917

  Copy full SHA for 6cc0917

Commits on Jul 8, 2024

• [Mosaic TPU] Add lowering for lax.erf_inv 32 bit

  ayaka14732

  committed
  1b196c7

  Copy full SHA for 1b196c7

Commits on Jul 5, 2024

• Update persistent compilation cache doc

  ayaka14732

  committed
  db32021

  Copy full SHA for db32021

Commits on Jul 4, 2024

• [Pallas TPU] Add lowering for lax.shift_right_arithmetic

  ayaka14732

  committed
  f013bd5

  Copy full SHA for f013bd5

• Clean up

  ayaka14732

  committed
  6c05aa2

  Copy full SHA for 6c05aa2

Commits on Jul 3, 2024

• Storing the last access time of a cache entry in a separate file

  ayaka14732

  committed
  934142d

  Copy full SHA for 934142d

Commits on Jun 27, 2024

• Improve compilation cache tests

  ayaka14732

  committed
  bc7addf

  Copy full SHA for bc7addf

Commits on Jun 18, 2024

• [LRUCache] Fix CI failure on Windows

  Show description for 79cc04a
  

  ayaka14732

  authored and

  jax authors

  committed
  79cc04a

  Copy full SHA for 79cc04a

Commits on Jun 11, 2024

• Implement LRU cache eviction for persistent compilation cache

  Show description for 1a3a15c
  

  ayaka14732

  and

  superbobry

  committed
  1a3a15c

  Copy full SHA for 1a3a15c

Commits on Jun 4, 2024

• Fix typo

  ayaka14732

  committed
  ef22b46

  Copy full SHA for ef22b46

Commits on Mar 12, 2024

• Fix typo in multi_process.md

  ayaka14732

  committed
  cf62573

  Copy full SHA for cf62573

Commits on Sep 21, 2023

• Fix typo in CHANGELOG.md

  ayaka14732

  authored
  74bc42e

  Copy full SHA for 74bc42e

Commits on Jun 22, 2023

• Fix typo: ConcretizationError -> ConcretizationTypeError

  ayaka14732

  authored
  feb34ce

  Copy full SHA for feb34ce

• Fix typo in documentation

  ayaka14732

  authored
  5da5804

  Copy full SHA for 5da5804

Commits on Mar 29, 2022

• [doc] Fix typo

  ayaka14732

  authored
  2799bb3

  Copy full SHA for 2799bb3

Commits on Mar 4, 2022

• [Doc] Add jaxlib build instructions for TPU

  Show description for ee2377b
  ayaka14732

  authored
  ee2377b

  Copy full SHA for ee2377b

Pagination

Previous

Next