atlas-ftag-tools==0.2.17
comet_ml>=3.49.7
h5py>=3.14.0
jsonargparse[argcomplete,signatures]>=4.32.0
lightning==2.5.3
lion-pytorch>=0.2.2
numpy>=2.2.6
onnx==1.16.2
onnxruntime==1.20.1
puma-hep==0.4.11
rich>=14.1.0
scipy>=1.15.3
torch==2.5.1
torchmetrics==1.5.2
torchvision==0.20.1
tqdm>=4.66.1
umami-preprocessing==0.2.7

[dev]
coverage>=7.10.6
ipykernel>=6.30.1
mypy>=1.18.1
pre-commit>=4.3.0
pydoclint>=0.7.3
pytest_notebook>=0.10.0
pytest-cov>=7.0.0
pytest-randomly>=4.0.1
pytest>=8.4.2
ruff>=0.13.0

[flash]
flash-attn>=2.5.7
ninja>=1.11.1.1
packaging
wheel

[muP]
aiobotocore==2.24.2
boto3==1.40.18
botocore==1.40.18
mup>=1.0.0
s3fs==2024.10.0
s3path>=0.5.0
tensorboard>=2.14.1
