Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Oct 3, 2025

• 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

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

  Show description for 77a2b28
  

  The tunix Authors

  committed
  77a2b28

  Copy full SHA for 77a2b28

• Pin Jax version to 0.7.1

  wang2yn84

  committed
  8c9b6cb

  Copy full SHA for 8c9b6cb

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

  Show description for 1c23df8
  

  The tunix Authors

  committed
  1c23df8

  Copy full SHA for 1c23df8

• Downgrad the Jax version to resolve performance issue Cloud TPU.

  wang2yn84

  committed
  aab457a

  Copy full SHA for aab457a

• Add Typed... types to ArrayLike.

  Show description for fd25133
  

  hawkinsp

  authored and

  The tunix Authors

  committed
  fd25133

  Copy full SHA for fd25133

Commits on Sep 30, 2025

• [Tunix] Reshard prompt and completion tokens to the `REFERENCE` mesh before computing reference log probabilities if needed. This is needed when rollout and reference are distributed.

  Show description for 2a9fe73
  

  lc5211

  authored and

  The tunix Authors

  committed
  2a9fe73

  Copy full SHA for 2a9fe73

• Enable generic creation of optax.optimizer, optax.learning_rate_schedule from cli

  Show description for 8be9c41
  

  sizhit2

  authored and

  The tunix Authors

  committed
  8be9c41

  Copy full SHA for 8be9c41

• Add min resources requirements for launch scripts to README

  Show description for ed8b737
  

  sizhit2

  authored and

  The tunix Authors

  committed
  ed8b737

  Copy full SHA for ed8b737

• Delete this notebook as it's redundant now. Prepare for the release.

  Show description for 520c7c0
  

  The tunix Authors

  committed
  520c7c0

  Copy full SHA for 520c7c0

• Make `huggingface_hub` visible to tunix

  Show description for b522969
  

  sizhit2

  authored and

  The tunix Authors

  committed
  b522969

  Copy full SHA for b522969

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

  Show description for 46d8fb8
  

  The tunix Authors

  committed
  46d8fb8

  Copy full SHA for 46d8fb8

• Bump up the version to v0.1.0

  wang2yn84

  committed
  991f276

  Copy full SHA for 991f276

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

  Show description for 16b7679
  

  The tunix Authors

  committed
  16b7679

  Copy full SHA for 16b7679

• Bump up the version to 0.1.0.dev1

  wang2yn84

  committed
  dfcd68e

  Copy full SHA for dfcd68e

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

  Show description for bb10d34
  

  The tunix Authors

  committed
  bb10d34

  Copy full SHA for bb10d34

• Fix the kaggle downloading in GRPO notebook.

  wang2yn84

  committed
  eb7c149

  Copy full SHA for eb7c149

• Rename Gemma config classes to ModelConfig for consistency.

  Show description for 09cb200
  

  abheesht17

  authored and

  The tunix Authors

  committed
  09cb200

  Copy full SHA for 09cb200

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

  Show description for 9794c90
  

  The tunix Authors

  committed
  9794c90

  Copy full SHA for 9794c90

• Adds prod and dev mode for tunix.

  wang2yn84

  committed
  1d8aaef

  Copy full SHA for 1d8aaef

• Fix Gemma loading

  Show description for f1acbcf
  

  abheesht17

  authored and

  The tunix Authors

  committed
  f1acbcf

  Copy full SHA for f1acbcf

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

  Show description for 7532814
  

  The tunix Authors

  committed
  7532814

  Copy full SHA for 7532814

Pagination

Previous

Next