| [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"] | |