pixi.toml

[workspace]
authors = ["Michael Booth <michael@databooth.com.au>"]
channels = ["conda-forge", "https://conda.modular.com/max-nightly"]
name = "fasterai-mojo"
platforms = ["osx-arm64", "linux-64"]
version = "0.1.0"

[activation]
# Point Mojo tools at this project's Mojo sources by default
env = { MOJO_PATH = "$PIXI_PROJECT_ROOT/mojo" }

[tasks]
mojo-version = "mojo --version"
gradient-check = "PYTHONPATH=python python python/experiments/gradient_check.py"
performance = "python python/experiments/performance_compare.py"
train = { cmd = "PYTHONPATH=python python python/experiments/train_compare.py" }

[dependencies]
mojo = "*"
# Max Python bindings
max = "*"
python = ">=3.13"
# Core numeric stack
numpy = "*"
# PyTorch + ecosystem (CPU by default)
pytorch = "*"
torchvision = "*"
torchaudio = "*"
# fastai for training pipeline
fastai = "*"