pixi.toml

[project]
name = "python"
version = "1.0.0"
description = "Python for Machine Learning"
authors = ["Jim Hargrave <jim.hargrave@strakergroup.com>"]
channels = ["conda-forge", "nvidia"]
platforms = ["linux-64"]

[dependencies]
jaxlib = "*=*cuda*"
jax = "*"
pip = "*"
keras = "*"
pydantic = "*"
python = "~=3.11"
numpy = "*"
pandas = "*"
scikit-learn = "*"
matplotlib = "*"
jupyter = "*"
pytest = "*"
ipython = "*"
coloredlogs = "*"
pylint = "*"
python-dotenv = "*"
ollama = "*"
joblib = ">=1.3.2,<1.4.0"

[pypi-dependencies]
torch = "*"
dspy-ai = { git = "https://github.com/stanfordnlp/dspy.git", extras = ["mongodb"]}
textgrad = { git = "https://github.com/zou-group/textgrad.git" }

[[tool.pixi.source]]
name = "torch"
url = "https://download.pytorch.org/whl/cu124"
packages = ["torch"]

[system-requirements]
cuda = "12.4"