Commits

Branch selector

 main

User selector

superbobry

Datepicker

All time

Commit History

Commits on Oct 10, 2025

• [pallas:sc] Allowed scalar loads and stores to/from VMEM

  Show description for ae0ea1d
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  ae0ea1d

  Copy full SHA for ae0ea1d

• [mosaic] Removed the deprecated `tpu.wait_dma` op

  Show description for e2a4c1a
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  e2a4c1a

  Copy full SHA for e2a4c1a

Commits on Oct 9, 2025

• [jaxlib] Generate .pyi files for the _jax extension via nanobind's stubgen

  Show description for b53b634
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  b53b634

  Copy full SHA for b53b634

• [pallas:mosaic_gpu] Sketched a decorator exposing Pallas Mosaic GPU kernels to PyTorch

  Show description for d0ea64e
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  d0ea64e

  Copy full SHA for d0ea64e

Commits on Oct 8, 2025

• [pallas:sc] Fixed calls to `api_util.debug_info`

  Show description for 0fafb83
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  0fafb83

  Copy full SHA for 0fafb83

Commits on Oct 7, 2025

• [pjrt] Do not ignore `shape` layout in `CommonPjRtClient::CreateViewOfDeviceBuffer`

  Show description for d0ecd2a
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  d0ecd2a

  Copy full SHA for d0ecd2a

• [jaxlib] Revived `PyTreeDef::FromNodeDataAndChildren`

  Show description for b4f2676
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  b4f2676

  Copy full SHA for b4f2676

• [pallas:mosaic_gpu] Allowed `scratch_shapes=` to be a mapping in `plgpu.kernel`

  Show description for 73095f0
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  73095f0

  Copy full SHA for 73095f0

Commits on Oct 6, 2025

• [pallas] Another attempt at migrating `pl.empty*` APIs to use `lax.empty`

  Show description for 349cbc2
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  349cbc2

  Copy full SHA for 349cbc2

• [jax2tf] Aligned the type of `polymorphic_shapes` with the contract in the docstring

  Show description for 40e6811
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  40e6811

  Copy full SHA for 40e6811

• [jaxlib] Updated lots of jaxlib nanobind wrappers to produce better stubs

  Show description for 4280f54
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  4280f54

  Copy full SHA for 4280f54

Commits on Oct 2, 2025

• [pallas:sc] `pltpu.async_copy` can now do dynamically-sized scatters and gathers

  Show description for ef9d852
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  ef9d852

  Copy full SHA for ef9d852

Commits on Oct 1, 2025

• [jaxlib] Improved the type annotations for various nanobind-exported APIs

  Show description for 89d2f21
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  89d2f21

  Copy full SHA for 89d2f21

• `nanobind_pywrap_extension` now accepts arbitrary kwargs

  Show description for e45312b
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  e45312b

  Copy full SHA for e45312b

Commits on Sep 30, 2025

• [pallas:sc] Marked all Pallas SC tests as ondemand

  Show description for bbc081b
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  bbc081b

  Copy full SHA for bbc081b

Commits on Sep 29, 2025

• undefined-variable is not a valid pytype error code

  Show description for 9574858
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  9574858

  Copy full SHA for 9574858

• Addressed an old TODO to remove `ABCMeta` from `basearray.Array`

  Show description for 40b8576
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  40b8576

  Copy full SHA for 40b8576

Commits on Sep 26, 2025

• [pallas:sc] Added a basic test using `pltpu.emit_pipeline`

  Show description for 8bc535c
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  8bc535c

  Copy full SHA for 8bc535c

Commits on Sep 25, 2025

• [pallas:sc] Added overloads for `parallel_loop` and slightly reformatted the docstring

  Show description for 85edead
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  85edead

  Copy full SHA for 85edead

• [state] Allowed refs as indexers

  Show description for 7d5a09a
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  7d5a09a

  Copy full SHA for 7d5a09a

• Bumped mypy to 1.18.2 and ruff to 0.13.1

  Show description for ac6ea95
  superbobry

  committed
  ac6ea95

  Copy full SHA for ac6ea95

Commits on Sep 24, 2025

• [pallas:sc] Simplified the `lax.bitcast_convert_type` test

  Show description for a02d6a7
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  a02d6a7

  Copy full SHA for a02d6a7

• [pallas:sc] Do not emit trivial bitcasts to the same dtype

  Show description for 62b5112
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  62b5112

  Copy full SHA for 62b5112

Commits on Sep 23, 2025

• [pallas:sc] Added support for int32 to `plsc.{pack,unpack}`

  Show description for 9f8e65a
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  9f8e65a

  Copy full SHA for 9f8e65a

• [pallas:sc] Added a libtpu version guard to `test_bitcast`

  Show description for 6f2aa1b
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  6f2aa1b

  Copy full SHA for 6f2aa1b

• Removed the no longer necessary jax-aval-named-shape deprecation

  Show description for fa3b7b9
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  fa3b7b9

  Copy full SHA for fa3b7b9

Commits on Sep 22, 2025

• [pallas:mosaic] Removed a stale version guard

  Show description for 094f643
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  094f643

  Copy full SHA for 094f643

• [pallas:sc] Removed an SC-specific lowering rule for `lax.bitcast_convert_type`

  Show description for ee83eff
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  ee83eff

  Copy full SHA for ee83eff

• [mlir] Removed unused traceback to location machinery

  Show description for d14604a
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  d14604a

  Copy full SHA for d14604a

Commits on Sep 19, 2025

• Added a note on the now removed `for_loop` API to the changelog

  Show description for 671aba1
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  671aba1

  Copy full SHA for 671aba1

• [pallas:sc] Allowed using `lax.reshape` in SC kernels

  Show description for c3b5cad
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  c3b5cad

  Copy full SHA for c3b5cad

• [pallas:sc] Allowed specifying tiling for `pl.run_scoped`-allocated refs

  Show description for 2fc1660
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  2fc1660

  Copy full SHA for 2fc1660

• [pallas:sc] Fixed the type of `sc.loop_unroll_factor`

  Show description for dc73a08
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  dc73a08

  Copy full SHA for dc73a08

• [pallas] Avoid making `pl.loop` bounds dynamic when it is not necessary

  Show description for 9092f5a
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  9092f5a

  Copy full SHA for 9092f5a

Commits on Sep 17, 2025

• [pallas] Added `mesh_axis_types` to `ShapedArrayWithMemorySpace`

  Show description for b463797
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  b463797

  Copy full SHA for b463797

Pagination

Previous

Next