Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Jul 22, 2025

• improve test coverage.

  Show description for daf08c6
  

  lc5211

  authored and

  copybara-github

  committed
  daf08c6

  Copy full SHA for daf08c6

• move sync_weight to rl_cluster

  Show description for 94b0048
  

  tianshub

  authored and

  copybara-github

  committed
  94b0048

  Copy full SHA for 94b0048

• [TunixW&B] Install `wandb` in qlora_demo

  Show description for 62a0ce6
  

  jshin1394

  authored and

  copybara-github

  committed
  62a0ce6

  Copy full SHA for 62a0ce6

• remove unnecessary comments.

  Show description for 37c06ae
  

  lc5211

  authored and

  copybara-github

  committed
  37c06ae

  Copy full SHA for 37c06ae

• bootstrap RL cluster

  Show description for 17651ea
  

  tianshub

  authored and

  copybara-github

  committed
  17651ea

  Copy full SHA for 17651ea

Commits on Jul 21, 2025

• [Tunix] Enable to log learning rate at each train step.

  Show description for 737287c
  

  sizhit2

  authored and

  copybara-github

  committed
  737287c

  Copy full SHA for 737287c

Commits on Jul 18, 2025

• Add Colab self-links to example notebooks.

  Show description for 0d1adc8
  

  Wei Wei

  authored and

  copybara-github

  committed
  0d1adc8

  Copy full SHA for 0d1adc8

• common rl trainer abstraction

  Show description for 9dff097
  

  tianshub

  authored and

  copybara-github

  committed
  9dff097

  Copy full SHA for 9dff097

• Simple heuristic to eliminate replicated all-gather.

  Show description for ee3a7c7
  

  Tunix Developers

  authored and

  copybara-github

  committed
  ee3a7c7

  Copy full SHA for ee3a7c7

Commits on Jul 17, 2025

• [TunixW&B] Create a unique wandb_run_name based on datetime.

  Show description for b7fcd9a
  

  jshin1394

  authored and

  copybara-github

  committed
  b7fcd9a

  Copy full SHA for b7fcd9a

• [TunixProfiler] Add invalid step number check in profiler.

  Show description for c1868cb
  

  jshin1394

  authored and

  copybara-github

  committed
  c1868cb

  Copy full SHA for c1868cb

• minor cleanup

  Show description for f6162e5
  

  tianshub

  authored and

  copybara-github

  committed
  f6162e5

  Copy full SHA for f6162e5

Commits on Jul 16, 2025

• stop grad for compute per token logps

  Show description for 5dd3f7c
  

  tianshub

  authored and

  copybara-github

  committed
  5dd3f7c

  Copy full SHA for 5dd3f7c

• [Tunix] Add test to showcase Tunix support learning rate schedule.

  Show description for 9762843
  

  sizhit2

  authored and

  copybara-github

  committed
  9762843

  Copy full SHA for 9762843

• Copybara import of the project:

  Show description for 7bcc7de
  

  cgarciae

  authored and

  copybara-github

  committed
  7bcc7de

  Copy full SHA for 7bcc7de

Commits on Jul 15, 2025

• [Tunix] Fix the issue that step_fn would compile twice due to incorrect sharding in optimizer

  Show description for fe728f2
  

  annyan09023

  authored and

  copybara-github

  committed
  fe728f2

  Copy full SHA for fe728f2

• Internal

  Show description for ddcc4e8
  

  abheesht17

  authored and

  copybara-github

  committed
  ddcc4e8

  Copy full SHA for ddcc4e8

Commits on Jul 14, 2025

• Disable the backward pass for inference worker.

  Show description for 15622fe
  

  Tunix Developers

  authored and

  copybara-github

  committed
  15622fe

  Copy full SHA for 15622fe

• fix the memory usage issue around per token log probs calculation

  Show description for 529266b
  

  Tunix Developers

  authored and

  copybara-github

  committed
  529266b

  Copy full SHA for 529266b

• Minor changes.

  Show description for b90dc62
  

  Tunix Developers

  authored and

  copybara-github

  committed
  b90dc62

  Copy full SHA for b90dc62

• Add back the resharding APIs for weights sync.

  Show description for fdd878d
  

  Tunix Developers

  authored and

  copybara-github

  committed
  fdd878d

  Copy full SHA for fdd878d

Commits on Jul 10, 2025

• bootstrap inference worker

  Show description for 70c8dae
  

  tianshub

  authored and

  copybara-github

  committed
  70c8dae

  Copy full SHA for 70c8dae

• [Tunix] Add data queue abstraction. Later we can add ser/de logic, ckpt, etc.

  Show description for 8ba907b
  

  annyan09023

  authored and

  copybara-github

  committed
  8ba907b

  Copy full SHA for 8ba907b

Commits on Jul 9, 2025

• Adds the llama3 1B model for development. Smaller model is faster to iterate.

  Show description for a9f1319
  

  Tunix Developers

  authored and

  copybara-github

  committed
  a9f1319

  Copy full SHA for a9f1319

Commits on Jul 8, 2025

• Adds a function to return a dummy input to the llama3 model.

  Show description for d4df30d
  

  Tunix Developers

  authored and

  copybara-github

  committed
  d4df30d

  Copy full SHA for d4df30d

Commits on Jul 3, 2025

• add Jax named_scope

  Show description for 938d4b5
  

  liudangyi

  authored and

  copybara-github

  committed
  938d4b5

  Copy full SHA for 938d4b5

• [Tunix] Replace grpo trainer with new grpo learner

  Show description for dbcd284
  

  Tunix Developers

  authored and

  copybara-github

  committed
  dbcd284

  Copy full SHA for dbcd284

Commits on Jul 2, 2025

• Bootstrap rollout worker abstraction

  Show description for f394e89
  

  tianshub

  authored and

  copybara-github

  committed
  f394e89

  Copy full SHA for f394e89

• [Tunix] Add resuming support for grpo learner

  Show description for 1d1b2a1
  

  Tunix Developers

  authored and

  copybara-github

  committed
  1d1b2a1

  Copy full SHA for 1d1b2a1

• [Tunix] Create async queue between sampler and trainer

  Show description for fea30d6
  

  Tunix Developers

  authored and

  copybara-github

  committed
  fea30d6

  Copy full SHA for fea30d6

Commits on Jul 1, 2025

• [Tunix] Fix some parity issue between grpo learner and grpo trainer

  Show description for 5c47c7b
  

  Tunix Developers

  authored and

  copybara-github

  committed
  5c47c7b

  Copy full SHA for 5c47c7b

Commits on Jun 30, 2025

• minor renaming

  Show description for b172d85
  

  tianshub

  authored and

  copybara-github

  committed
  b172d85

  Copy full SHA for b172d85

• [Tunix] Improve performance

  Show description for 1792cdf
  

  Tunix Developers

  authored and

  copybara-github

  committed
  1792cdf

  Copy full SHA for 1792cdf

Commits on Jun 27, 2025

• internal changes.

  Show description for 5cd7554
  

  Tunix Developers

  authored and

  copybara-github

  committed
  5cd7554

  Copy full SHA for 5cd7554

• remove test only output field

  Show description for 72fa710
  

  tianshub

  authored and

  copybara-github

  committed
  72fa710

  Copy full SHA for 72fa710

Pagination

Previous

Next