Repository navigation

Commits

Commits on Oct 22, 2026
- Support PyTorch builds without distributed support (#110)
  anutosh491
  authored
  a83f8ca
Commits on Oct 11, 2026
- Reset PRNG key placement in distributed_array tutorial (#109)
  wdhongtw
  authored
  d95b0a5
Commits on Sep 17, 2026
- Use lax.div in avg_pool so integer inputs stay integer (#104)
  benSepanski
  authored
  ef79eed
Commits on Sep 11, 2026
- Backport torch.ops.aten.qr after PyTorch switch to torch.linalg.qr (#108)
  ManfeiBai
  authored
  08982ee
- Add support VHLO target versioning and serialization in export (#107)
  ManfeiBai
  authored
  1205739
Commits on Aug 30, 2026
- Bump torch from 2.8.0+cpu to 2.13.0 in the pip group across 1 directory (#103)
  dependabot[bot]
  authored
  7d0f8ab
Commits on Aug 7, 2026
Commits on Jul 3, 2026
- Remove `torch.cholesky` overrides (#99)
  malfet
  authored
  ac59e3d
Commits on Jun 17, 2026
- bump version to 0.0.14 (#100)
  anthonsu
  authored
  eeffe37
Commits on Jun 15, 2026
- feat: static shape jaten ops for multimodal embedding (#96)
  anthonsu
  authored
  027f4f1
Commits on May 19, 2026
- bump version to 0.0.13. (#92)
  wdhongtw
  authored
  663604c
Commits on May 11, 2026
- Fix torchax tensor flatten bug on empty tensors (#93)
  cychiuak
  authored
  7605a49
Commits on Apr 28, 2026
- Track more functions as TENSOR_CONSTRUCTORS (#91)
  wdhongtw
  authored
  9ce5f2d
Commits on Apr 23, 2026
- Add typing friendly transformation interfaces (#90)
  wdhongtw
  authored
  ed7d9a3
Commits on Apr 10, 2026
- Thread-safety for building global environment (#89)
  wdhongtw
  authored
  8f957d1
Commits on Apr 8, 2026
- Simplify property manipulation in Environment (#87)
  wdhongtw
  authored
  9ad2dcd
- Correct heading in README for TPU only topic (#88)
  wdhongtw
  authored
  1eafcff
- Merging two CONTRIBUTING documents (#86)
  wdhongtw
  authored
  54e1aaa
Commits on Mar 18, 2026
- Remove unused code in op registration (#78)
  wdhongtw
  authored
  060df56
Commits on Mar 17, 2026
- Support torch.nn.pad_sequence in torchax. (#84)
  yixinshi
  authored
  670498d
- Replace deprecated assert_allclose with assert_close (#83)
  wdhongtw
  authored
  994d120
Commits on Mar 9, 2026
- Upgrade GitHub Actions to latest versions (#81)
  salmanmkc
  authored
  67dcfdb
- Upgrade GitHub Actions for Node 24 compatibility (#80)
  salmanmkc
  authored
  21fbd1b
Commits on Mar 4, 2026
- Upgrade usage of deprecated PyTorch API (#76)
  wdhongtw
  authored
  b6fd9b3
Commits on Mar 3, 2026
- Apply new format for ruff tool (#77)
  wdhongtw
  authored
  1887bbd
Commits on Feb 5, 2026
- bump version to 0.0.12. (#74)
  QiliangCui
  authored
  6a4024f
Commits on Feb 2, 2026
- Remove dead code from source (#73)
  wdhongtw
  authored
  3f0550c
Commits on Jan 30, 2026
- Implement the `extract_jax` function using `JittableModule` (#75)
  kasper0406
  authored
  1436daf
- Fix few op related issues: (#69)
  qihqi
  authored
  1a5db91
Commits on Jan 28, 2026
- Correct the dev package to use YYYYMMDD naming (#72)
  wdhongtw
  authored
  cdd45f7
- Reduce package dependency and warning (#71)
  wdhongtw
  authored
  15ed343
Commits on Jan 26, 2026
- Resolve unintentional registration of add operator (#70)
  wdhongtw
  authored
  320ff06
Commits on Jan 20, 2026
- Add simple LoRA example using huggingface peft (#66)
  jack8558
  authored
  85c7ab9

Branch selector

User selector

Datepicker

Commit history

Commits on Oct 22, 2026

Commits on Oct 11, 2026

Commits on Sep 17, 2026

Commits on Sep 11, 2026

Commits on Aug 30, 2026

Commits on Aug 7, 2026

Commits on Jul 3, 2026

Commits on Jun 17, 2026

Commits on Jun 15, 2026

Commits on May 19, 2026

Commits on May 11, 2026

Commits on Apr 28, 2026

Commits on Apr 23, 2026

Commits on Apr 10, 2026

Commits on Apr 8, 2026

Commits on Mar 18, 2026

Commits on Mar 17, 2026

Commits on Mar 9, 2026

Commits on Mar 4, 2026

Commits on Mar 3, 2026

Commits on Feb 5, 2026

Commits on Feb 2, 2026

Commits on Jan 30, 2026

Commits on Jan 28, 2026

Commits on Jan 26, 2026

Commits on Jan 20, 2026