maxtext.experimental package

Subpackages

• maxtext.experimental.agent package
  • Subpackages
    • maxtext.experimental.agent.ckpt_conversion_agent package
      • Subpackages
      • Submodules
• maxtext.experimental.rl package
  • Submodules
    • maxtext.experimental.rl.grpo_input_pipeline module
      • SingleHostDataLoader
      • preprocessing_pipeline()
      • make_hf_train_iterator()
      • create_data_iterator()
    • maxtext.experimental.rl.grpo_trainer module
    • maxtext.experimental.rl.grpo_utils module