LICENSE
README.md
pyproject.toml
google_tunix.egg-info/PKG-INFO
google_tunix.egg-info/SOURCES.txt
google_tunix.egg-info/dependency_links.txt
google_tunix.egg-info/requires.txt
google_tunix.egg-info/top_level.txt
tunix/__init__.py
tunix/cli/config.py
tunix/cli/grpo_main.py
tunix/cli/peft_main.py
tunix/cli/utils/model.py
tunix/cli/utils/reward.py
tunix/distillation/__init__.py
tunix/distillation/distillation_trainer.py
tunix/distillation/feature_extraction/__init__.py
tunix/distillation/feature_extraction/pooling.py
tunix/distillation/feature_extraction/projection.py
tunix/distillation/feature_extraction/sowed_module.py
tunix/distillation/strategies/__init__.py
tunix/distillation/strategies/attention.py
tunix/distillation/strategies/base_strategy.py
tunix/distillation/strategies/feature_pooling.py
tunix/distillation/strategies/feature_projection.py
tunix/distillation/strategies/logit.py
tunix/examples/data/math_dataset.py
tunix/examples/data/translation_dataset.py
tunix/generate/base_sampler.py
tunix/generate/beam_search.py
tunix/generate/sampler.py
tunix/generate/tokenizer_adapter.py
tunix/generate/utils.py
tunix/generate/vllm_sampler.py
tunix/models/safetensors_loader.py
tunix/models/gemma/model.py
tunix/models/gemma/params.py
tunix/models/gemma/params_safetensors.py
tunix/models/gemma/sampler.py
tunix/models/gemma3/model.py
tunix/models/gemma3/params.py
tunix/models/gemma3/params_safetensors.py
tunix/models/llama3/model.py
tunix/models/llama3/params.py
tunix/models/qwen2/model.py
tunix/models/qwen2/params.py
tunix/models/qwen3/model.py
tunix/models/qwen3/params.py
tunix/oss/utils.py
tunix/rl/common.py
tunix/rl/reshard.py
tunix/rl/rl_cluster.py
tunix/rl/rl_learner.py
tunix/rl/trainer.py
tunix/rl/utils.py
tunix/rl/experimental/agentic/agents/agent_types.py
tunix/rl/experimental/agentic/agents/base_agent.py
tunix/rl/experimental/agentic/agents/tool_agent.py
tunix/rl/experimental/agentic/environments/base_environment.py
tunix/rl/experimental/agentic/environments/tool_environment.py
tunix/rl/experimental/agentic/parser/tool_parser/gemini_parser.py
tunix/rl/experimental/agentic/parser/tool_parser/qwen_parser.py
tunix/rl/experimental/agentic/parser/tool_parser/tool_parser_base.py
tunix/rl/experimental/agentic/parser/tool_parser/tool_parser_registry.py
tunix/rl/experimental/agentic/prompts/prompt_template.py
tunix/rl/experimental/agentic/rewards/reward.py
tunix/rl/experimental/agentic/rewards/reward_types.py
tunix/rl/experimental/agentic/tools/base_tool.py
tunix/rl/experimental/agentic/tools/calculator_tool.py
tunix/rl/experimental/agentic/tools/tool_manager.py
tunix/rl/experimental/agentic/trajectory/trajectory_collect_engine.py
tunix/rl/grpo/grpo_helpers.py
tunix/rl/grpo/grpo_learner.py
tunix/rl/inference/inference_worker.py
tunix/rl/ppo/ppo_helpers.py
tunix/rl/ppo/ppo_learner.py
tunix/rl/queue/data_queue.py
tunix/rl/rollout/base_rollout.py
tunix/rl/rollout/vanilla_rollout.py
tunix/rl/rollout/vllm_rollout.py
tunix/sft/checkpoint_manager.py
tunix/sft/hooks.py
tunix/sft/inflight_throttler.py
tunix/sft/metrics_logger.py
tunix/sft/peft_trainer.py
tunix/sft/profiler.py
tunix/sft/progress_bar.py
tunix/sft/sharding_utils.py
tunix/sft/system_metrics_calculator.py
tunix/sft/utils.py
tunix/sft/dpo/dpo_trainer.py
tunix/tests/test_common.py