Metadata-Version: 2.1
Name: salt-ml
Version: 0.7
Summary: Multimodal and Multiclass Machine Learning for High Energy Physics
License: MIT
Project-URL: Homepage, https://gitlab.cern.ch/atlas-flavor-tagging-tools/algorithms/salt
Project-URL: Issue Tracker, https://gitlab.cern.ch/atlas-flavor-tagging-tools/algorithms/salt/-/issues
Requires-Python: <3.12,>=3.10
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: pytest==7.4.2
Requires-Dist: pytest-cov==4.1.0
Requires-Dist: comet_ml==3.35.3
Requires-Dist: tqdm==4.66.1
Requires-Dist: rich==13.6.0
Requires-Dist: pre-commit==3.5.0
Requires-Dist: ruff==0.2.2
Requires-Dist: h5py==3.10.0
Requires-Dist: numpy==1.24.2
Requires-Dist: torch==2.2.1
Requires-Dist: lightning==2.2.0
Requires-Dist: jsonargparse[argcomplete,signatures]==4.32.0
Requires-Dist: torchmetrics==1.2.1
Requires-Dist: onnx==1.16.0
Requires-Dist: onnxruntime==1.15.1
Requires-Dist: atlas-ftag-tools==0.2.2
Requires-Dist: scipy==1.12.0
Provides-Extra: mup
Requires-Dist: mup==1.0.0; extra == "mup"
Requires-Dist: boto3==1.28.17; extra == "mup"
Requires-Dist: s3fs==2023.9.2; extra == "mup"
Requires-Dist: s3path==0.5.0; extra == "mup"
Requires-Dist: tensorboard==2.14.1; extra == "mup"
Provides-Extra: flash
Requires-Dist: wheel; extra == "flash"
Requires-Dist: packaging; extra == "flash"
Requires-Dist: ninja==1.11.1.1; extra == "flash"
Requires-Dist: flash-attn==2.5.7; extra == "flash"

[![docs](https://img.shields.io/badge/info-documentation-informational)](https://ftag-salt.docs.cern.ch/)

# Salt

This is the home of Salt, a framework for training multi-model and multi-task models models in the style of GN2.

Documentation is available [here](https://ftag-salt.docs.cern.ch/).
