Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Oct 7, 2025

• Support 'generate_with_chat_template' functionality for agentic rl

  Show description for 8e16652
  

  hgao327

  authored and

  The tunix Authors

  committed
  8e16652

  Copy full SHA for 8e16652

• Tool code update

  Show description for 4abf16b
  

  hgao327

  authored and

  The tunix Authors

  committed
  4abf16b

  Copy full SHA for 4abf16b

• Remove prompt_template.py

  Show description for e743611
  

  hgao327

  authored and

  The tunix Authors

  committed
  e743611

  Copy full SHA for e743611

Commits on Oct 6, 2025

• Merge pull request #515 from google:lance-fix

  Show description for 6e3682d
  

  The tunix Authors

  committed
  6e3682d

  Copy full SHA for 6e3682d

• [Tunix] Allow specifying a different data type for the rollout model.

  Show description for 1b5f925
  

  lc5211

  authored and

  The tunix Authors

  committed
  1b5f925

  Copy full SHA for 1b5f925

• Fix the front page pip install command.

  wang2yn84

  committed
  6c148bd

  Copy full SHA for 6c148bd

• Update agentic APIs

  Show description for eb8481f
  

  hgao327

  authored and

  The tunix Authors

  committed
  eb8481f

  Copy full SHA for eb8481f

• Merge pull request #481 from google:lance-description

  Show description for 6d57c93
  

  The tunix Authors

  committed
  6d57c93

  Copy full SHA for 6d57c93

• Update descriptions.

  wang2yn84

  committed
  2761922

  Copy full SHA for 2761922

• Disable setting specific profiler options on Pathways backend.

  Show description for d20c245
  

  SurbhiJainUSC

  authored and

  The tunix Authors

  committed
  d20c245

  Copy full SHA for d20c245

• Merge pull request #501 from chethanuk:add-gemma3-270m-support

  Show description for e00ceec
  

  The tunix Authors

  committed
  e00ceec

  Copy full SHA for e00ceec

Commits on Oct 4, 2025

• Add Gemma3-270M model configuration support

  Show description for b6528ab
  chethanuk

  committed
  b6528ab

  Copy full SHA for b6528ab

Commits on Oct 3, 2025

• Merge pull request #447 from google:tunix_grpo_test

  Show description for cf656b9
  

  The tunix Authors

  committed
  cf656b9

  Copy full SHA for cf656b9

• Pass device indexes to vLLM

  Show description for 9f5a8ce
  

  The tunix Authors

  committed
  9f5a8ce

  Copy full SHA for 9f5a8ce

• Merge pull request #464 from google:tp_kv_head_repeat

  Show description for 928c3d4
  

  The tunix Authors

  committed
  928c3d4

  Copy full SHA for 928c3d4

• Fix type errors within agents/

  Show description for e5a21e6
  

  hgao327

  authored and

  The tunix Authors

  committed
  e5a21e6

  Copy full SHA for e5a21e6

• Merge pull request #495 from google:lance-fix-ckpt-path

  Show description for 3caecdc
  

  The tunix Authors

  committed
  3caecdc

  Copy full SHA for 3caecdc

• Fix the colab that can't find the final checkpoint. Cleanup the intermediate checkpoint directory.

  wang2yn84

  committed
  1d36319

  Copy full SHA for 1d36319

• Add TPU tests workflow with GRPO integration test

  Show description for 774cc91
  mydatascience

  committed
  774cc91

  Copy full SHA for 774cc91

• Fix reward function bug

  Show description for 2eb4184
  

  hgao327

  authored and

  The tunix Authors

  committed
  2eb4184

  Copy full SHA for 2eb4184

• Fix the slow kaggle image issue. Investigated everything else, the mesh setup is the culprit. Mesh of (1, 8) is 2x faster than (2, 4). Mesh (1, 4) is at the same level, about 1.5s per training ite…

  Show description for 3a06226
  

  The tunix Authors

  committed
  3a06226

  Copy full SHA for 3a06226

• Code update

  Show description for 804c284
  

  jenriver

  authored and

  The tunix Authors

  committed
  804c284

  Copy full SHA for 804c284

Commits on Oct 2, 2025

• Code update

  Show description for f8baab9
  

  tianshub

  authored and

  The tunix Authors

  committed
  f8baab9

  Copy full SHA for f8baab9

• new reward functions and unit tests

  Show description for 6642af6
  

  hgao327

  authored and

  The tunix Authors

  committed
  6642af6

  Copy full SHA for 6642af6

• Enable setting default param dtype for Gemma 3 model

  Show description for 8ebfb84
  

  jeffcarp

  authored and

  The tunix Authors

  committed
  8ebfb84

  Copy full SHA for 8ebfb84

• Move tests under experimental/agentic from tpu test session to cpu session.

  Show description for 5ecad84
  

  hgao327

  authored and

  The tunix Authors

  committed
  5ecad84

  Copy full SHA for 5ecad84

• Add utility functions for agentic RL.

  Show description for 8647e07
  

  hgao327

  authored and

  The tunix Authors

  committed
  8647e07

  Copy full SHA for 8647e07

• Add chat template parsers for agentic models.

  Show description for 188449c
  

  hgao327

  authored and

  The tunix Authors

  committed
  188449c

  Copy full SHA for 188449c

• Create a new list container type to support both flax<=0.11.2 and >=0.12.0.

  Show description for aa354f1
  

  The tunix Authors

  committed
  aa354f1

  Copy full SHA for aa354f1

Commits on Oct 1, 2025

• add remat options to model

  Show description for f585e7a
  

  tianshub

  authored and

  The tunix Authors

  committed
  f585e7a

  Copy full SHA for f585e7a

• Merge pull request #474 from google:v0.1

  Show description for f722f74
  

  The tunix Authors

  committed
  f722f74

  Copy full SHA for f722f74

• Adds experimental pre-shard to Pathways on Cloud `experimental_reshard`

  Show description for 3f831d4
  

  lukebaumann

  authored and

  The tunix Authors

  committed
  3f831d4

  Copy full SHA for 3f831d4

• Relax the jax version requirement to get a working Kaggle image.

  wang2yn84

  committed
  8fc04b9

  Copy full SHA for 8fc04b9

• update internal grpo notebook

  Show description for 7fe4207
  

  tianshub

  authored and

  The tunix Authors

  committed
  7fe4207

  Copy full SHA for 7fe4207

• add a comment for version pinning

  Show description for 4f0b077
  

  tianshub

  authored and

  The tunix Authors

  committed
  4f0b077

  Copy full SHA for 4f0b077

Pagination

Previous

Next