atlas-ftag-tools==0.3.3
comet_ml>=3.49.7
filelock>=3.12
h5py>=3.14.0
jsonargparse[argcomplete,signatures]>=4.32.0
lightning==2.6.1
lion-pytorch>=0.2.2
numpy>=2.2.6
onnx==1.21.0
onnxruntime==1.25.0
onnxscript
puma-hep==0.5.3
rich>=14.1.0
scipy>=1.15.3
torch==2.12.0
torchmetrics==1.8.2
tqdm>=4.66.1
umami-preprocessing==0.3.1

[flash]
flash-attn==2.8.3
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
