Tunix Files

API Change

This release fixes a critical issue introduced in v0.1.4 that prevented correct functionality. Users are strongly recommended to upgrade to v0.1.5.

# old:
rl_trainer = GrpoLearner(
  grpo_config=grpo_config,
)
# new:
rl_trainer = GrpoLearner(
  algo_config=grpo_config,
)

What's Changed

• Remove grpo helper. by @copybara-service[bot] in https://github.com/google/tunix/pull/771
• Fix the GitHub source links in example notebooks on dpo, grpo and qlora by @rajasekharporeddy in https://github.com/google/tunix/pull/775
• adding support for cns file downloads in tunix cli by @copybara-service[bot] in https://github.com/google/tunix/pull/762
• Developing on v0.1.5 now by @wang2yn84 in https://github.com/google/tunix/pull/776
• Replace grpo_config with algo_config while calling GRPOLearner in GRPO Demo notebook by @rajasekharporeddy in https://github.com/google/tunix/pull/778
• Lazy load transformers by @copybara-service[bot] in https://github.com/google/tunix/pull/779
• Fix first_micro_batch_rollout_time by @copybara-service[bot] in https://github.com/google/tunix/pull/783

Full Changelog: https://github.com/google/tunix/compare/v0.1.4...v0.1.5