maxtext.experimental.rl package

Submodules

• maxtext.experimental.rl.grpo_input_pipeline module
  • SingleHostDataLoader
    • SingleHostDataLoader.global_mesh
    • SingleHostDataLoader.dataloader
    • SingleHostDataLoader.local_iterator
    • SingleHostDataLoader.reset()
  • preprocessing_pipeline()
  • make_hf_train_iterator()
  • create_data_iterator()
• maxtext.experimental.rl.grpo_trainer module
• maxtext.experimental.rl.grpo_utils module