[project]
name = "craftax-remdm-planner"
version = "0.1.0"
description = "Add your description here"
readme = "README.md"
requires-python = ">=3.12"
dependencies = [
    "chex>=0.1.91",
    "craftax>=1.5.0",
    "distrax>=0.1.7",
    "flax>=0.12.6",
    "huggingface-hub>=1.9.1",
    "jax>=0.9.2",
    "matplotlib>=3.10.8",
    "numpy>=2.4.4",
    "optax>=0.2.8",
    "orbax>=0.1.9",
    "orjson>=3.11.8",
    "polars>=1.39.3",
    "pyyaml>=6.0.3",
    "wandb>=0.25.1",
]

[project.optional-dependencies]
cuda = ["jax[cuda13]>=0.9.2"]