Wheelodex » Projects » salt-ml

salt-ml

View on PyPI — Reverse Dependencies (0)

0.8 salt_ml-0.8-py3-none-any.whl

Wheel Details

Project: salt-ml
Version: 0.8
Filename: salt_ml-0.8-py3-none-any.whl
Download: [link]
Size: 161351
MD5: 521b8fc3a63beb73966163026748d652
SHA256: a0e56bc88de64b18751878f3ede1c3b7f9ae91e97133ec68253dc4648b138fab
Uploaded: 2024-09-23 13:08:19 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt · entry_points.txt

METADATA

Metadata-Version: 2.1
Name: salt-ml
Version: 0.8
Summary: Multimodal and Multiclass Machine Learning for High Energy Physics
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
License: MIT
Requires-Python: <3.12,>=3.10
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.4)
Requires-Dist: scipy (==1.12.0)
Requires-Dist: lion-pytorch (==0.2.2)
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"
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
Provides-Extra: mup
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 297 characters]

WHEEL

Wheel-Version: 1.0
Generator: setuptools (75.1.0)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
salt/__init__.py sha256=WY9HbkgoxJlUCfEaMASIAo3wFTh3K3UXHty6LsjolMg 20
salt/main.py sha256=k080VQ4T6Qt7tFOwFnuDZeIEKf5x9Yyr9Zx7kDr99v0 849
salt/modelwrapper.py sha256=f9xuhxr_6D543XMsvfUIJreYP6lqiwr945VMQ03wgi0 9642
salt/stypes.py sha256=YrnZOfdpYNFrpanHeQEKc0oHAyiyXFMtpJr3T18INoc 249
salt/callbacks/__init__.py sha256=8HXScDIfoE3ZSLpZ44Q7u34hpG3XEoVAR2OuKE8gS44 427
salt/callbacks/checkpoint.py sha256=GKMs38DZeshe9Z945H-HgJ5I2EKlhYCzrkaBvch25Dk 1400
salt/callbacks/maskformer_metrics.py sha256=XaEtWrmQ33m8yXpzfV-pW4x3CliiHmBikNA8CUdm97s 7171
salt/callbacks/performancewriter.py sha256=lJx1sZHct9kgxKokUKuklDrj4N3MZjN-x_OeoSic28g 2496
salt/callbacks/predictionwriter.py sha256=nAsTG3qSAUYAN_387R50T14FtaqBEcSPl-tbYA1o6I8 10686
salt/callbacks/saveconfig.py sha256=fpnG63ZcruT7xO8u_5GqXolyJIHZK5rl39ChHhAOAOs 10146
salt/configs/Baseline_Xbb.yaml sha256=YJg7y79Q8zLzerJ-A8cGh-BcLg-fgBZ3YXuqfLvyaRc 2087
salt/configs/DL1.yaml sha256=GQXgNElj07qLsy_t8ZB0Ivg38VOn-a8H9Uze-3rp2bg 1694
salt/configs/Dipz.yaml sha256=N8qy59Z7QtCkVRFlyPDuzLB6eFyCvfCqUnBDjkWuY1k 2317
salt/configs/GATv2.yaml sha256=_nOB_Q54gIIPIeJIxjzukG_sPZY7QEoNSJbQr6EiDt0 378
salt/configs/GN1.yaml sha256=aEnQVcAsWjdq4Fvf9HumJv_Sxr_OoVeOCNnlwJZJX-w 3677
salt/configs/GN2.yaml sha256=MKbhvvOB-bG8X120ZlhbHEX9oxSNmZ1iBRXOpZcOhyg 3815
salt/configs/GN2X.yaml sha256=04U7eomV8_zHM3M0X_p2uZxufm2n_gPqZOaG9r0i7qc 4061
salt/configs/GN2XE.yaml sha256=uMLp_bulmpSCB1n5N1WmBuuosd8TETfIaZAzDSaMsl8 5015
salt/configs/GN2_charded_neutral_loose_aux.yaml sha256=SCMI3KFd4sQSV5iIphdu4nS9mbTsm9Gg422VjsXXHOM 4941
salt/configs/GN2_extended.yaml sha256=DKCAkunD5lWlyNtNFYP8ZOnixKqU_wDjSrZbv2DmnPc 3808
salt/configs/GN2_muP.yaml sha256=gK_Qy1EYvRwEG4F8Q2sFI9LjhfoHsb2vVyRaDNxNTNA 4142
salt/configs/GN2_tracks_neutral_CA.yaml sha256=LpMy1lfOXq00tGpJrNA6o23Z4xz5VLAISkpsniCDeUs 3234
salt/configs/GN2_tracks_neutral_SA_aux.yaml sha256=mAbHdsj27ILskbiXjqyKkdfaUEkG0mv78mUWer-5yxU 4099
salt/configs/GN2emu.yaml sha256=wNyPZEUICCaOGHwVMpD-GaaLRitPGebowE8Oo2KKmd4 5680
salt/configs/GN3.yaml sha256=L3SerBbpUR88Oex2qcO47UUtLW_PYd9LjGrCcP60xIY 3627
salt/configs/MaskFormer.yaml sha256=ITCK8Ja_UrO8g5eOfA31bQcVZE8Aq7dS41nyjXBRWeg 6196
salt/configs/PLITel.yaml sha256=WOF4YYTpZUnkgwLrcotRpb5JrH9KptlLV84fpkyGC5Y 4127
salt/configs/SubjetXbb.yaml sha256=7j81O8YjDCWOE8xRSIhBnInoXqgTzzfRJx7PWt6Ffy0 3041
salt/configs/TPLTmu.yaml sha256=zId7UfIzXbIMFVzWM4rwzSQJT5_KdRSy6yH8Y9WWzPM 4724
salt/configs/base.yaml sha256=UXCC3Qud4_10rvCP3x7vAxdWFE-JV0qd8n2rmFaVr5M 1416
salt/configs/dips.yaml sha256=GLSua7MHM5cM-G8YGlcufM7wwOGceFfX12COzbDOIMA 1750
salt/configs/encoder-v2.yaml sha256=fPPj_FKPjVAPZHAYgryX17_pIz7EyXw84TmNeeFWy94 269
salt/configs/event_classifier.yaml sha256=xjn6q7QfYjZudZx_1afnznBz-_nRiFdPuqDtgvII5H8 2828
salt/configs/flow.yaml sha256=B8TRbw8yXhVQioiTtwZGP7xrJNaQJqKdlUZBtXwEXeg 2053
salt/configs/hitz.yaml sha256=HawimbSq3VSJsd6X2KCqz8N9Q5w7WmIGmN5563K6-cc 2664
salt/configs/nan_regression.yaml sha256=Ge5_rFHnk0iVHyXQ8q3kRRXqKIZY7H7gJ0RJAr8o9mA 2287
salt/configs/regression.yaml sha256=4srFCyJF1IDnGEJKv8mRx8as1wqEGr70_hup-a_9uZw 3394
salt/configs/regression_gaussian.yaml sha256=V54H0xlKv12UJqtjM--vbK8oe27SUEy4Rxc48wqTtk0 1758
salt/configs/tutorial.yaml sha256=UGOSfDaFuERh2DUDgGrrqQ-Apq1GM89rbeayQJ05v1g 3255
salt/data/__init__.py sha256=9t-G8rIojq29YI6NUA97EilNgRytfsT5avJ0xUcJu8w 148
salt/data/datamodules.py sha256=uCniu4oq_d6DzcuhjGltWCgbJjuaZ-RphMAITgVGd2s 6173
salt/data/datasets.py sha256=anxXw-4j8dqXBGy82yVaVVJGABV8vsD2ssiYryqnb_g 13489
salt/data/edge_features.py sha256=cx6Xi5JLC0h3uL5nePEgp6T_qCvjUueDp2mZpw34PK8 3639
salt/data/samplers.py sha256=1Nof_uY0IIEpnsWVGudAtDHKtSBeA8m36eJUi_wThP0 1875
salt/metrics/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
salt/models/__init__.py sha256=85RS3IH6-tNdPPFJZvv_fEBbLPZ2ZBnr4bf48Pir0WU 1773
salt/models/attention.py sha256=qoNyw5xvNhZWJUIuAC4Wi6h1-ntiRRu0wYnIKUPJPzk 13669
salt/models/dense.py sha256=u-uNahf9DflIfQl3q0-z3gNiHbTdC2-NNceqGAg8aD4 3572
salt/models/featurewise.py sha256=5k1fb4x-KnT6h6iPEk1bVhh5LArTz9MkLyZU5AAszuQ 2773
salt/models/initnet.py sha256=ttIzJKz0-eG2f866vhZo619PqHTivj8jhn2D7qp6tnU 3580
salt/models/inputnorm.py sha256=RE_hxipBRQAZVBuZWMTKLiE9MPz8LC40O5Cbxfky0oM 3203
salt/models/layernorm.py sha256=oAuwn5TkS1UGJQBFWAM-5pS6KOIae7AnKHkgalX1vXc 628
salt/models/maskformer.py sha256=h9vukUloOXIc1moe7PKeC0d6WFXJRw8YIVJDMcmB8Iw 11088
salt/models/maskformer_loss.py sha256=0c95AE5cYxMNotDmuzQGlhmncm8Wf-4jIxEeZlDjNBU 9562
salt/models/matcher.py sha256=8lQn4MM10RAae_cbICwGO52WNLRuIlbii4Yc2SA4BtM 8047
salt/models/pooling.py sha256=7g_8UlAE-uSprFCP80bzRhASLwbN8VvMuJP2Y0AxGlg 5221
salt/models/posenc.py sha256=knVPom_Q8MwXJIkVOLxWYh9xX8neD5Rdcuaepsnb9X0 2681
salt/models/r21xbb.py sha256=jhidtGLJJglPxiJ_dy9nQ9wvTU_59ZlR90mcUmglxls 1134
salt/models/saltmodel.py sha256=fMBHHY0ajkBoU80ovej6xztosTgIQJtv8Y4GWaeIfm4 9346
salt/models/task.py sha256=-XZfMAVqHYEQGtWYQB2khgtp2E3AcC1jS0-PBCh3sPU 22669
salt/models/transformer.py sha256=KpJOvzslU5V7sA8oJ8_s90LnCa0308sC9UVfP7ErwY8 13591
salt/models/transformer_v2.py sha256=zWLCsoWNLeG9P8pq4AhcpyTPLiIti4ialCN4taUJsxw 24910
salt/onnx/check.py sha256=h_Zi9awueTSt8GUZcufyjQCHvoHnSGQdI_9kiHIX_0o 4177
salt/onnx/to_onnx.py sha256=sWdnDROVUKRDDt2rFyf3-XqIzPQY-5jQcd6v7yiPVxY 16374
salt/submit/condor_handler.py sha256=W6Z2x_o3QDxAV60mAtPSaC_e6jKEggS66hdlpYJtytk 3632
salt/submit/slurm_handler.py sha256=GfYv30YRIihRCj_1CpIooi4hJYb5NooPYSeXQlUwoXA 2604
salt/submit/submit_htcondor.py sha256=ovX8PgcjHH-8g1vBb2ajUlvxKr8jGcoLhbSS3TlEHWw 2209
salt/submit/submit_slurm.py sha256=Alr0ILVNkDMY4HDIMq2TSr1mREXRTb-phSHEAHv4WRw 6481
salt/tests/test_datasets.py sha256=f_BmAebNcvKO3U2cWZdQcheKkZWs2uMjBeQPlaqhJg4 919
salt/tests/test_edges.py sha256=BEAmaBCqnEhtPlnKhsicunmp_WtcYuXOIIwEBEIermY 3856
salt/tests/test_featurewise.py sha256=Bsq0lu46GKLYYC6tri7aahWrt6pdWZMXEzCy7eLkHu8 837
salt/tests/test_initnet.py sha256=bh5z5o9bAAbYNkbNupkXO_Kp_7E2W7oKkk4wBrPAMhg 1224
salt/tests/test_masks.py sha256=RQJV1uQv5LuJGtN0L1LF_ns17dws6PwYAKW3k9szuE0 2314
salt/tests/test_models.py sha256=ZvIYOLfNEnOA732v39fNsmH2nqQL0N68mIeJvQ3Ia08 7643
salt/tests/test_pipeline.py sha256=jeJzjbRvwUvAQUOTLjpYvTuKLpEVv5p-K4DSg4ospT0 10147
salt/tests/test_posenc.py sha256=MT27iwq-8kTD4Cfn71HBvksDDDCWGxFBU_qrMr8Su2Q 1485
salt/tests/test_transformerv2.py sha256=Mv-5UXylTCuavy9cjYUhptFIVWEyo9alUDcciJ9NxeU 14363
salt/tests/test_union_find.py sha256=rklyGSqh0UNP93FNLPWpK4dQyLUhjr-qW00lWEq6dkY 2689
salt/tests/test_utils.py sha256=95HVrL9Wobj_ovURq8UykYEilFV6Gu1UbrGO0Yjnc_k 4391
salt/utils/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
salt/utils/array_utils.py sha256=6l6uqYo5sYslSyx-AcwTfQ7ZjG-LSHyp2sazgQJYRB4 1497
salt/utils/benchmarking.py sha256=GUhEQisFXVAPjbhUvUFrDatHms5nY31sMi7BlexTVVw 4808
salt/utils/class_names.py sha256=COQkGyCFhfFZTfqtJ8hb0KuVWqkd8VfGS8d0cQYJDuc 344
salt/utils/clean_logs.py sha256=T1HpEsFTT5Ye1fQ89HbRjn-v-QCf3qDldks2QXRlRtc 1013
salt/utils/cli.py sha256=P2qoh_KRH3jjo-Aq_ZVkTuByxLYkoCQb7u9HPHukeiI 13138
salt/utils/compare_models.py sha256=1ux8pR27Hm_xtb7gfOr3yr-PrmPAoU2kua3H7Z0gsfE 4384
salt/utils/configs.py sha256=8RuWUGaFD3OAnghUs1Lvgc64lB-UdOEEq9smETEnRKU 3384
salt/utils/file_utils.py sha256=P8_r0Xx93_llD84LonYu5HsH8KMtTprqcfQLdpexzzg 6019
salt/utils/get_onnx_metadata.py sha256=Bw-4kBOSCpkSPmJZkp53h7vgwIlscpmERvPwdyyosnU 1217
salt/utils/inputs.py sha256=w9vAdP3HfwoPPVbqAnYfZ1TC4CzQCGfKr9hTIAORHKs 12644
salt/utils/mask_utils.py sha256=A9Plx3LY34E2JJokWwcqDpXgNokOX0okGk7IK50prAU 8420
salt/utils/repair_ckpt.py sha256=WH-JLfHxj8Vgt8OGdVOto1f_cOtQGmajGSWSwlbsZjo 1498
salt/utils/scalers.py sha256=Clga9Lgv8nqgxBGeb5tz_TkEIkwnOsbe5lCf8DeZpqg 3260
salt/utils/tensor_utils.py sha256=pT852V_Gd73slzr9Rl_4PC3Qgy9abd8wStZSKZbGLJk 4772
salt/utils/union_find.py sha256=VvMwBR0GogjqqrCQ8mdA4Jc1ZE76bad5pHFKJ7D5HMc 4704
salt/utils/muP_utils/configuration_muP.py sha256=oyQZC3UqmiiEVGIAuAvycN-U7MB42xY-Ru2WthHRWE0 16427
salt/utils/muP_utils/functions_check_muP.py sha256=vgu6-EWVtL9mXJ4sITa9UNMqEbLuTKZcVMj9tWXF4Tc 27742
salt/utils/muP_utils/generateModel.py sha256=ci7N2033BjKeofTKqchpCBcSujx0rPUiCF7t07uQB1I 1486
salt/utils/muP_utils/main_muP.py sha256=Kt4ytH2k4748di7xDccdJpJc37bZFDD206eDz9kbF-0 5013
salt/utils/muP_utils/smaller_it.py sha256=TxtjwE9xEocWzPZCkX91Su2jLByv0MbXXlRUyEpEZ7w 1907
salt_ml-0.8.dist-info/LICENSE sha256=xhKDC68K3tkxStAz7NqUflkVqfvXR_Zw4nsWbsisXq8 1052
salt_ml-0.8.dist-info/METADATA sha256=LJj-er6sTJz8o3oGalPm6-XEBc10f7Lj7mZ95SWmGgM 1796
salt_ml-0.8.dist-info/WHEEL sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ 91
salt_ml-0.8.dist-info/entry_points.txt sha256=U3e3HaUskBtZQMhevO_XNhXDZyxP4DNVpe9USQZIVEE 317
salt_ml-0.8.dist-info/top_level.txt sha256=BqSzl_xwRerFJ8Kuw_9GzuXOMAFsdg8jjCbnmQL-Z7Y 5
salt_ml-0.8.dist-info/RECORD — —

top_level.txt

salt

entry_points.txt

[console_scripts]
compare_models = salt.utils.compare_models:main
download_S3 = salt.utils.file_utils:download_from_S3
get_onnx_metadata = salt.utils.get_onnx_metadata:main
repair_ckpt = salt.utils.repair_ckpt:main
salt = salt.main:main
setup_muP = salt.utils.muP_utils.main_muP:main
to_onnx = salt.onnx.to_onnx:main