[project]
name = "jax-optuna-neural-ode"
version = "0.1.0"
description = "Add your description here"
readme = "README.md"
requires-python = ">=3.13"
dependencies = [
    "diffrax>=0.7.0",
    "equinox>=0.13.0",
    "jax>=0.7.1",
    "joblib>=1.5.2",
    "loguru>=0.7.3",
    "matplotlib>=3.10.6",
    "numpy>=2.3.2",
    "optax>=0.2.5",
    "optuna>=4.5.0",
    "pandas>=2.3.2",
    "psutil>=7.1.0",
    "pydantic>=2.11.7",
    "scienceplots>=2.2.0",
    "scipy>=1.16.1",
    "torch>=2.9.1",
    "torchdiffeq>=0.2.5",
]