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: 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'
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'

[![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/).
