Skip to content

README.md

Latest commit

 

History

History
31 lines (21 loc) · 772 Bytes

README.md

File metadata and controls

31 lines (21 loc) · 772 Bytes

Looking for the generalized information-gathering work? If you arrived here from the paper Generalized Information Gathering Under Dynamics Uncertainty, see the generalized branch.

MAX

Experimental JAX-based model-based RL library. Re-implements TDMPC2 with an emphasis on functional/pure-style code. Expect sharp edges.

Installation

From source

git clone <repository-url>
cd max
pip install -e .

Dependencies

Core (installed via pip):

  • jax, jaxlib - Core framework
  • flax - Neural network definitions
  • optax - Optimizers
  • numpy, scipy, matplotlib

Optional:

  • wandb - Experiment tracking (pip install -e ".[wandb]")

License

MIT License