Pull requests: google/tunix
Pull requests list
remove _MeshBound wrapper
ready-to-submit
#2268
opened Sep 17, 2026 by
copybara-service
Bot
Loading…
Group TPU CI jobs by container image to halve per-PR slice demand.
ready-to-submit
#2267
opened Sep 17, 2026 by
copybara-service
Bot
Loading…
Deliver config that was accepted but never passed
#2265
opened Sep 17, 2026 by
entrpn
Collaborator
Loading…
6 tasks done
Adds
.to_atif() projection methods to Step and TrajectoryMetadata in trajectory.py. This allows Tunix RL subclasses (TunixAgentStep, TunixEnvStep, TunixTrajectoryMetadata) to project cleanly to base ATIF models by packing subclass-specific fields into extra.
#2263
opened Sep 16, 2026 by
copybara-service
Bot
Loading…
* Added
QwenVtcChatTemplateParser subclassing QwenChatTemplateParser to format system/user turns with Qwen ChatML control tokens (<|im_start|>, <|im_end|>) and prefill <reasoning>\n into the assistant generation prompt (_init_generation_prompt) and _parse_assistant.
#2262
opened Sep 16, 2026 by
copybara-service
Bot
Loading…
[docker] Pin jax/jaxlib/libtpu to the ABI tpu-inference resolved
#2261
opened Sep 16, 2026 by
khatwanimohit
Collaborator
Loading…
6 tasks done
Pin sglang-jax to LKG commit in TPU CI workflows.
#2260
opened Sep 16, 2026 by
copybara-service
Bot
Loading…
Support profiling options in MaxText trainer and launchers.
#2257
opened Sep 16, 2026 by
copybara-service
Bot
Loading…
ci: halve per-PR TPU slice demand and add job timeouts
ready-to-submit
#2254
opened Sep 16, 2026 by
shz-google
Loading…
[Tunix] Support DeepSWE evaluation pipeline with MaxText and vLLM on TPU
#2249
opened Sep 16, 2026 by
susanbao
Collaborator
Loading…
6 tasks done
Add design and architecture documentation (
experimental/docs/clip_ratio_and_raw_length_metrics.md) detailing how response-length truncation (generation/completions/clip_ratio) and raw response length (generation/completions/{mean,max,min}_raw_length) metrics are computed across the producer-consumer rollout and trainer pipeline:
#2245
opened Sep 16, 2026 by
copybara-service
Bot
Loading…
Add CPU preemption drill integration test for AgenticRLLearner.
#2242
opened Sep 16, 2026 by
copybara-service
Bot
Loading…
[Tunix] Make GradientAccumulator an immutable pytree value instead of an nnx.Module.
#2241
opened Sep 16, 2026 by
copybara-service
Bot
Loading…
[Tunix] Replace AbstractTrainer.per_token_logps with a generic fwd_only method.
#2235
opened Sep 15, 2026 by
copybara-service
Bot
Loading…
Add support for multi-turn MoE router replay by passing an engine-level prompt offset.
#2230
opened Sep 15, 2026 by
copybara-service
Bot
Loading…
Log the learning rate when a gradient transformation precedes the optimizer.
#2223
opened Sep 15, 2026 by
copybara-service
Bot
Loading…
Implement CodeActAgent and NeMo Gym integration for OpenHands in DeepSWE:
ready-to-submit
#2222
opened Sep 14, 2026 by
copybara-service
Bot
Loading…
Adds background worker (
AsyncSqlWriter) and model serialization codecs
#2208
opened Sep 11, 2026 by
copybara-service
Bot
Loading…
Refactors
async_writer.py into a storage-agnostic, generic asynchronous queue
#2207
opened Sep 11, 2026 by
copybara-service
Bot
Loading…
Adds SQL Trajectory Store writer implementation (
SqlTrajectoryStore) for Tunix.
#2206
opened Sep 11, 2026 by
copybara-service
Bot
Loading…
Support sequence packing in sub-batch checkpointing for the agentic RL learner.
#2195
opened Sep 11, 2026 by
copybara-service
Bot
Loading…
ProTip!
Mix and match filters to narrow down what you’re looking for.