Pull requests: google/tunix
Pull requests list
Update the distributed GSM8K GRPO example to use the real GSM8K dataset and VTC reward
#2037
opened Aug 29, 2026 by
copybara-service
Bot
Loading…
Support
enable_continuous_profiling option in the Tunix JAX profiler wrapper
#2036
opened Aug 29, 2026 by
youchunni
Loading…
5 tasks done
MaxText/Tunix: Expose enable_continuous_profiling CLI flag in MaxText and Tunix profiler to allow saving >2GB profile as xplane.riegeli
#2035
opened Aug 28, 2026 by
copybara-service
Bot
Loading…
Switch the first batch of sandbox "wait" flag to be True so we make sure the first batches always getting pre-warmed.
#2028
opened Aug 28, 2026 by
copybara-service
Bot
Loading…
Make weight synchronization optional in distributed GRPO
#2024
opened Aug 28, 2026 by
copybara-service
Bot
Loading…
Add GSM8KEnv and GSM8KAgent for Tunix agentic RL
#2022
opened Aug 28, 2026 by
andytwigg
Collaborator
Loading…
Add MaxText and Raiden optional dependencies to Dockerfile
#2013
opened Aug 27, 2026 by
SurbhiJainUSC
Collaborator
Loading…
6 tasks done
feat(rollout): add VllmSamplerAdapter with Raiden weight sync support
#2012
opened Aug 27, 2026 by
SurbhiJainUSC
Collaborator
Loading…
6 tasks done
Add Pyrefly type ignore in FrozenLake data.
#2011
opened Aug 27, 2026 by
copybara-service
Bot
Loading…
Support
base_agentic_config.yaml relative path in CLI config loader.
#2008
opened Aug 27, 2026 by
copybara-service
Bot
Loading…
Fix Gemma 4 KV cache initialization and add params dependency.
#2007
opened Aug 27, 2026 by
copybara-service
Bot
Loading…
Add FrozenLake and PPO GSM8k configs to
xm_launch data dependencies.
#2006
opened Aug 27, 2026 by
copybara-service
Bot
Loading…
Use
int type for num_input_tokens in SamplingState.
#2005
opened Aug 27, 2026 by
copybara-service
Bot
Loading…
[Tunix] Replace legacy TrainExample with RLTrainerPayload in experimental orchestrator and workers.
#2000
opened Aug 27, 2026 by
copybara-service
Bot
Loading…
feat: add YAML template for multihost vllm sampler.
#1995
opened Aug 26, 2026 by
copybara-service
Bot
Loading…
[Tunix] Shift checkpoint restoration control from the trainer to the RL orchestrator.
#1994
opened Aug 26, 2026 by
copybara-service
Bot
Loading…
feat(examples): integrate MaxText trainer backend into distributed GSM8K example
#1991
opened Aug 26, 2026 by
SurbhiJainUSC
Collaborator
Loading…
6 tasks done
feat(weight_sync): Raiden weight synchronization with progressive memory reclamation
#1990
opened Aug 26, 2026 by
SurbhiJainUSC
Collaborator
Loading…
6 tasks done
- Merge trajectory lineage contexts into hardware-sized microbatch payloads using LineageContext.merge().
#1986
opened Aug 26, 2026 by
copybara-service
Bot
Loading…
Rollout Ingestion, Auto-Stamping & Worker Telemetry for Distributed RL Lineage.
#1985
opened Aug 26, 2026 by
copybara-service
Bot
Loading…
[Tunix] Inject RaidenWeightSyncDelegate into Vanilla and InprocessVllm Sampler Adapters.
#1975
opened Aug 25, 2026 by
copybara-service
Bot
Loading…
fix(scripts): fix checkpoint sharding restore, precompile CLI flag, and rollout sync in grpo demo
#1962
opened Aug 23, 2026 by
Adityakk9031
Loading…
feat(peft): add BEFT (Bias-Elevation Fine-Tuning) and generalized PEFT support
#1961
opened Aug 23, 2026 by
Adityakk9031
Loading…
Add TODO, construct the WeightSynchronizer inside the trainer
#1958
opened Aug 22, 2026 by
copybara-service
Bot
Loading…
ProTip!
Updated in the last three days: updated:>2026-08-25.