nnx-ppo

Tutorials

  • Quickstart
  • Custom networks 1: composing with containers
  • Custom networks 2: graph networks
  • Custom networks 3: writing your own StatefulModule

Reference

  • Training parameters
  • Logging
  • Checkpointing
  • The rollout_extras channel
  • Batching contract
  • Randomness
  • Containers
  • Pytree utilities
  • Delay and Normalizer placement
  • PPOAdapter reference
  • Recording activations

API Reference

  • Networks
  • Algorithms
  • Wrappers
  • JAX Dataclass Utility
nnx-ppo
  • Search

© Copyright 2026, Emil Wärnberg.

Built with Sphinx using a theme provided by Read the Docs.