Pull requests: google/tunix
Pull requests list
Implement graceful shutdown for the rollout worker node.
#2166
opened Sep 9, 2026 by
copybara-service
Bot
Loading…
Add ML Diagnostics scalar backend for Tunix RL.
#2165
opened Sep 9, 2026 by
copybara-service
Bot
Loading…
Suppress new Pyrefly findings before upgrading to v1.2.0
#2164
opened Sep 9, 2026 by
copybara-service
Bot
Loading…
Support custom
kv_cache_size in base RL pipeline.
#2163
opened Sep 9, 2026 by
copybara-service
Bot
Loading…
Expose DeepSWE controls in distributed workers.
#2162
opened Sep 9, 2026 by
copybara-service
Bot
Loading…
Add standalone distributed DeepSWE components.
#2161
opened Sep 9, 2026 by
copybara-service
Bot
Loading…
Support per-request distributed episode limits.
#2160
opened Sep 9, 2026 by
copybara-service
Bot
Loading…
support for executing multiple optimizer updates (mini-batches) within a single global training step (full batch)
#2159
opened Sep 9, 2026 by
copybara-service
Bot
Loading…
[Logging] Wire --debug flag to log rollout sampled responses and evaluations
#2155
opened Sep 9, 2026 by
YixuanWang-99
Collaborator
Loading…
[Deployment] Parameterize JobSet manifests and minimal k8s_launcher for multi-host RL
#2152
opened Sep 9, 2026 by
YixuanWang-99
Collaborator
Loading…
[Raiden Weight Sync 6/6] MaxText trainer config plumbing and MoE dimension padding
#2151
opened Sep 9, 2026 by
YixuanWang-99
Collaborator
Loading…
[Raiden Weight Sync 5/6] MoE 128-lane weight interleaving for TPU GMM layout
#2150
opened Sep 9, 2026 by
YixuanWang-99
Collaborator
Loading…
[Raiden Weight Sync 4/6] Manage cache lifecycle and prefix caching in RaidenWeightSyncDelegate
#2149
opened Sep 9, 2026 by
YixuanWang-99
Collaborator
Loading…
[Raiden Weight Sync 3/6] Extract RaidenDestinationWeightSyncMixin for rollout samplers
#2148
opened Sep 9, 2026 by
YixuanWang-99
Collaborator
Loading…
[Raiden Weight Sync 2/6] FFI transport selection, preflight mismatch check, and host-stage normalization
#2147
opened Sep 9, 2026 by
YixuanWang-99
Collaborator
Loading…
[Raiden Weight Sync 1/6] Support multi-host discovery and multi-axis sharding in RaidenHandler
#2146
opened Sep 9, 2026 by
YixuanWang-99
Collaborator
Loading…
[tunix] Add Trajectory Store construction gating and lifecycle management
#2140
opened Sep 8, 2026 by
copybara-service
Bot
Loading…
Add Poisson preemption simulation tools and benchmark timing instrumentation for sub_batch.
#2138
opened Sep 8, 2026 by
copybara-service
Bot
Loading…
[Tunix] Add support for sequence packing in GRPO training.
#2135
opened Sep 8, 2026 by
copybara-service
Bot
Loading…
Gate jax[tpu] dependency behind a Linux platform marker
#2132
opened Sep 6, 2026 by
howl-anderson
Loading…
feat(generate): add jax-inference sampling backend and math evaluation support
ready-to-submit
#2130
opened Sep 4, 2026 by
andytwigg
Collaborator
Loading…
feat(rl): measure, mask and correct sampler-vs-trainer divergence
#2129
opened Sep 4, 2026 by
entrpn
Collaborator
Loading…
6 tasks
ProTip!
Mix and match filters to narrow down what you’re looking for.