[workspace]
authors = ["Uday Phalak <udayphlk@gmail.com>"]
channels = ["conda-forge", "pytorch"]
name = "continuous-thought-machines"
platforms = ["osx-arm64", "linux-64"]
version = "0.1.0"

[dependencies]
python = "3.12.*"
pytorch = { version = "*", channel = "pytorch" }
torchvision = { version = "*", channel = "pytorch" }
numpy = "<2.0"
matplotlib = "*"
seaborn = "*"
tqdm = "*"
opencv = "*"
imageio = "*"
scikit-learn = "*"
umap-learn = "*"
python-dotenv = "*"
gymnasium = "*"
datasets = "*"
huggingface_hub = "*"
safetensors = "*"
ffmpeg = "*"
accelerate = ">=1.12.0,<2"

[pypi-dependencies]
autoclip = "*"
minigrid = "*"