Commits

Branch selector

 main

User selector

All users

All time

Commit History

Commits on Apr 28, 2026

• Track more functions as TENSOR_CONSTRUCTORS (#91)

  Show description for 9ce5f2d
  wdhongtw

  authored
  9ce5f2d

  Copy full SHA for 9ce5f2d

Commits on Apr 23, 2026

• Add typing friendly transformation interfaces (#90)

  Show description for ed7d9a3
  wdhongtw

  authored
  ed7d9a3

  Copy full SHA for ed7d9a3

Commits on Apr 10, 2026

• Thread-safety for building global environment (#89)

  Show description for 8f957d1
  wdhongtw

  authored
  8f957d1

  Copy full SHA for 8f957d1

Commits on Apr 8, 2026

• Simplify property manipulation in Environment (#87)

  Show description for 9ad2dcd
  wdhongtw

  authored
  9ad2dcd

  Copy full SHA for 9ad2dcd

• Correct heading in README for TPU only topic (#88)

  Show description for 1eafcff
  wdhongtw

  authored
  1eafcff

  Copy full SHA for 1eafcff

• Merging two CONTRIBUTING documents (#86)

  wdhongtw

  authored
  54e1aaa

  Copy full SHA for 54e1aaa

Commits on Mar 18, 2026

• Remove unused code in op registration (#78)

  Show description for 060df56
  wdhongtw

  authored
  060df56

  Copy full SHA for 060df56

Commits on Mar 17, 2026

• Support torch.nn.pad_sequence in torchax. (#84)

  Show description for 670498d
  yixinshi

  authored
  670498d

  Copy full SHA for 670498d

• Replace deprecated assert_allclose with assert_close (#83)

  wdhongtw

  authored
  994d120

  Copy full SHA for 994d120

Commits on Mar 9, 2026

• Upgrade GitHub Actions to latest versions (#81)

  Show description for 67dcfdb
  salmanmkc

  authored
  67dcfdb

  Copy full SHA for 67dcfdb

• Upgrade GitHub Actions for Node 24 compatibility (#80)

  Show description for 21fbd1b
  salmanmkc

  authored
  21fbd1b

  Copy full SHA for 21fbd1b

Commits on Mar 4, 2026

• Upgrade usage of deprecated PyTorch API (#76)

  Show description for b6fd9b3
  wdhongtw

  authored
  b6fd9b3

  Copy full SHA for b6fd9b3

Commits on Mar 3, 2026

• Apply new format for ruff tool (#77)

  Show description for 1887bbd
  wdhongtw

  authored
  1887bbd

  Copy full SHA for 1887bbd

Commits on Feb 5, 2026

• bump version to 0.0.12. (#74)

  Show description for 6a4024f
  QiliangCui

  authored
  6a4024f

  Copy full SHA for 6a4024f

Commits on Feb 2, 2026

• Remove dead code from source (#73)

  Show description for 3f0550c
  wdhongtw

  authored
  3f0550c

  Copy full SHA for 3f0550c

Commits on Jan 30, 2026

• Implement the `extract_jax` function using `JittableModule` (#75)

  Show description for 1436daf
  kasper0406

  authored
  1436daf

  Copy full SHA for 1436daf

• Fix few op related issues: (#69)

  Show description for 1a5db91
  qihqi

  authored
  1a5db91

  Copy full SHA for 1a5db91

Commits on Jan 28, 2026

• Correct the dev package to use YYYYMMDD naming (#72)

  Show description for cdd45f7
  wdhongtw

  authored
  cdd45f7

  Copy full SHA for cdd45f7

• Reduce package dependency and warning (#71)

  Show description for 15ed343
  wdhongtw

  authored
  15ed343

  Copy full SHA for 15ed343

Commits on Jan 26, 2026

• Resolve unintentional registration of add operator (#70)

  Show description for 320ff06
  wdhongtw

  authored
  320ff06

  Copy full SHA for 320ff06

Commits on Jan 20, 2026

• Add simple LoRA example using huggingface peft (#66)

  jack8558

  authored
  85c7ab9

  Copy full SHA for 85c7ab9

Commits on Jan 7, 2026

• Add a doc for working with impure functions (#64)

  qihqi

  authored
  d45984c

  Copy full SHA for d45984c

Commits on Dec 23, 2025

• Add support for all in-place bitwise operators (#63)

  Show description for 79d339e
  wdhongtw

  authored
  79d339e

  Copy full SHA for 79d339e

Commits on Dec 22, 2025

• Remove option for using flash attention (#61)

  qihqi

  authored
  94b5af9

  Copy full SHA for 94b5af9

Commits on Dec 15, 2025

• Solve deprecation warning since jax 0.8.0 (#60)

  Show description for bfb1f5b
  wdhongtw

  authored
  bfb1f5b

  Copy full SHA for bfb1f5b

Commits on Dec 5, 2025

• bump version to 0.0.11

  Show description for 13ce870
  QiliangCui

  committed
  13ce870

  Copy full SHA for 13ce870

• add torch.ops.aten.unfold and torch.ops.aten.masked_scatter_ (#59)

  yuyanpeng-google

  authored
  7d3d4ba

  Copy full SHA for 7d3d4ba

Commits on Dec 3, 2025

• Add suport for fp4 (#57)

  kyuyeunk

  authored
  bf7d342

  Copy full SHA for bf7d342

• Misc doc update: (#58)

  Show description for 506a47b
  qihqi

  authored
  506a47b

  Copy full SHA for 506a47b

Commits on Dec 1, 2025

• Refactor test_access_config_thread to be consistent with others (#52)

  jialei777

  authored
  43b3733

  Copy full SHA for 43b3733

• Enable GKE/xpk tpu-v7 run with torchtitan example (#54)

  jialei777

  authored
  d87924f

  Copy full SHA for d87924f

Commits on Nov 18, 2025

• Update torchtitan dockerfile with latest torchax dependency

  jialei777

  authored
  a19e18b

  Copy full SHA for a19e18b

Commits on Nov 17, 2025

• ini (#46)

  jialei777

  authored
  470a95b

  Copy full SHA for 470a95b

• add ruff and fix (#48)

  Show description for 28f662c
  ShitongWang

  authored
  28f662c

  Copy full SHA for 28f662c

• Add release workflow

  qihqi

  committed
  d6d37b7

  Copy full SHA for d6d37b7