[project]
name = "diffusion-points"
version = "0.1.0"
description = "Add your description here"
readme = "README.md"
requires-python = ">=3.11"
dependencies = [
    "distrax>=0.1.5",
    "einops>=0.8.1",
    "flax>=0.10.6",
    "jax[cuda12]>=0.6.0",
    "numpy>=2.2.5",
    "orbax>=0.1.9",
    "seaborn>=0.13.2",
    "tqdm>=4.67.1",
    "tyro>=0.9.19",
    "wandb>=0.19.10",
]