franken

View on PyPIReverse Dependencies (0)

0.3.3 franken-0.3.3-py3-none-any.whl

Wheel Details

Project: franken
Version: 0.3.3
Filename: franken-0.3.3-py3-none-any.whl
Download: [link]
Size: 134364
MD5: 2bbde287c0361f1d4205e31cb62995a2
SHA256: 66806650c5206d5dbe625fe56217b492f20369a4a90d55334444e5359fefde83
Uploaded: 2025-03-28 14:55:48 +0000

dist-info

METADATA

Metadata-Version: 2.4
Name: franken
Version: 0.3.3
Summary: Franken fine-tuning scheme for ML potentials
Author: Giacomo Meanti, Luigi Bonati, Pedro Juan Buigues Jorro
Author-Email: Pietro Novelli <pietronvll[at]gmail.com>
Project-Url: Homepage, https://github.com/CSML-IIT-UCL/mlpot_transfer
Project-Url: Documentation, https://github.com/CSML-IIT-UCL/mlpot_transfer
Keywords: franken,molecular dynamics,potentials
Classifier: Development Status :: 4 - Beta
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: MIT License
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Requires-Python: >=3.10
Requires-Dist: ase
Requires-Dist: e3nn
Requires-Dist: hydra-core
Requires-Dist: numpy
Requires-Dist: omegaconf
Requires-Dist: psutil
Requires-Dist: requests
Requires-Dist: scipy
Requires-Dist: torch (>=2.4.0)
Requires-Dist: tqdm
Requires-Dist: cupy; extra == "cuda"
Requires-Dist: black (~=24.0); extra == "develop"
Requires-Dist: pre-commit; extra == "develop"
Requires-Dist: pytest; extra == "develop"
Requires-Dist: ruff; extra == "develop"
Requires-Dist: fairchem-core; extra == "fairchem"
Requires-Dist: mace-torch; extra == "mace"
Requires-Dist: sevenn; extra == "sevenn"
Provides-Extra: cuda
Provides-Extra: develop
Provides-Extra: fairchem
Provides-Extra: mace
Provides-Extra: sevenn
Description-Content-Type: text/markdown
License-Expression: MIT
License-File: LICENSE.txt
[Description omitted; length: 3000 characters]

WHEEL

Wheel-Version: 1.0
Generator: hatchling 1.27.0
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
franken/__init__.py sha256=pDs_-KM6-ZbPlBrR5BFQ2EKS7fMf_1RRn4OyfV4RM2U 163
franken/test_patch.py sha256=dWTfGWNcnQD5Lkyb2K84E899zC0QfcjIihwFmkapIMk 1729
franken/autotune/__init__.py sha256=-V5W2KdNJ8RAHAxAYZ2kteyQ8aTlSQ4I7yZrwXIzWMc 128
franken/autotune/__main__.py sha256=fn3jAHC8dKX1HQydjlofnj4Snvd939uMab67PbGLoP4 103
franken/autotune/evaluate.py sha256=jkoqQVq8M-8hX5CDkquMM2wON6yYPnHAxhksKHti8aE 5373
franken/autotune/script.py sha256=3xz9E1FYIi0VAeXhXvRldVipv8wI_FA-zWiF0EdQuws 12255
franken/autotune/configs/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
franken/autotune/configs/evaluate.yaml sha256=fWRB0XKP-IPSSM18GLdo3h1aulRFI2q3mgmSg3n3Cmo 875
franken/autotune/configs/master.yaml sha256=4Z2VpH3rE5vX8VmUiTbNGqb5BWXbS9l6ge4Ir9bJ3ls 227
franken/autotune/configs/hydra/job_logging/custom.yaml sha256=YVI_c6hLdyNYxALTg_ZSOyuX2b4FkIbLbKQxlVpcmU8 92
franken/autotune/configs/paths/begato.yaml sha256=ctsW4INhILG78tdGYgP7yr_sjP9oJFygKByrOBtbEXU 53
franken/autotune/configs/paths/default.yaml sha256=OOlujorslnP4w2mGRTCCv9f7yqTJt4dZ5iPD9vhhyGM 275
franken/autotune/configs/paths/edgar.yaml sha256=6JVXzHLK5XaQfkZy9VjutArMtV_3eRxRDmQ5Lnv10ZE 62
franken/autotune/configs/paths/franklin.yaml sha256=Jem8yw0U5e_WuCOME0IYjYwIRD-OSDHNsLDrnKVLCnM 71
franken/autotune/configs/paths/leonardo.yaml sha256=tN5rll5Cjsb_7li_XFqX2mDzGf5bA3AOd5W1dQCzsVk 100
franken/autotune/configs/paths/leonardo_IIT24_AtomSim.yaml sha256=vmFk1zq3Z9AACZrmd3LJZXMZwd2cJKluh3MEgIfa-mg 74
franken/autotune/configs/paths/leonardo_sample_complexity.yaml sha256=P0AHTG3dD8AlpZDr-PGkJaWLxZ6Xh42poPb7PPXIH3c 306
franken/autotune/configs/preset/default.yaml sha256=xPrViJM4st2ijYGg0DUH8ARq8PZITv-A4JuqYAqshU4 1921
franken/autotune/configs/preset/gaussian-MACE-L0.yaml sha256=6cU1rFqDGVvvKtRHtzLy9sVbL0dJnPxbhnq5v0QMEIk 316
franken/autotune/configs/preset/gaussian-SchNet-All.yaml sha256=bdGdq-aWZvanPWyMnlJlgL1tHVIiZSPPXtrTcfYlVdE 329
franken/autotune/configs/preset/gaussian-SevenNet0.yaml sha256=B3oRNrYjw79UOoMckTYLp6FB8thApPY1DRjAH7cAbYQ 366
franken/autotune/configs/preset/linear-MACE-L0.yaml sha256=OpcC_CMlq5J1BblgpbsrqLvRqjEahSR8gRImGvZXZFA 311
franken/autotune/configs/preset/linear-SchNet-All.yaml sha256=5l5yNE4KZtACfpx6X5zAf1isESl_tihvc7kQUE1vlbA 324
franken/autotune/configs/preset/linear-SevenNet0.yaml sha256=SOVcml-8pEIOP1PYyiAQZikABNS1DAnTYrXPDr0GfpE 361
franken/autotune/configs/preset/poly-MACE-L0.yaml sha256=5RWIyJ8xc8Nc_WL8iRBO5Z7XQj0hsfT0cdI-ODAWl1Q 314
franken/autotune/configs/preset/poly-SchNet-All.yaml sha256=rzo9cmMaw-8L-IDmshVU9aX_0l8CIg9zMBgpdF7dtd0 326
franken/backbones/__init__.py sha256=9ZIGPHzYcbC5_29uNefI6xqcJP9yFhXX9yu8Qzd79R0 91
franken/backbones/cli.py sha256=K56KfcDGeHnhkQ8qeSc_bP7y8hYWrAMtOCYc_qh_l-4 3060
franken/backbones/registry.json sha256=-OtkS2BG1gzsm4gMRuFi8eArfUhTd8gwHx3Cda4-XLE 23971
franken/backbones/utils.py sha256=Dc_5V_3yZ1izuKA97GG2CruWv4pl4INSV_R2E3zIn-Q 6581
franken/backbones/wrappers/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
franken/backbones/wrappers/common_patches.py sha256=1YeGvHMBPkI8Q_7WmA2cAaybDDFmwIBmml-sVUiXH2k 1138
franken/backbones/wrappers/fairchem_schnet.py sha256=sJYsrAgmWunHyqhbsEGgg2SSY7OzsrCcQrEd0ZT6-Ss 4351
franken/backbones/wrappers/mace_wrap.py sha256=9and9cDoI5BQU7T_P82-JnYsRNBJrea4rJ4Cwil0PDE 4087
franken/backbones/wrappers/sevenn.py sha256=cRjmxvm2KEYcXbmy921e6RcTe-xhJLG7NblepUzjDGI 5717
franken/calculators/ase_calc.py sha256=Qzs8Nxy5U6ImCohEdORg7oLtrwOngEkPrurkKLVW6wk 4079
franken/calculators/lammps.py sha256=PVGTrF5QdzHmZfLMrTpK9-hWB5bcOve9ITRNtFB23Lk 3635
franken/data/__init__.py sha256=blzlX9dmDqHXKUPbck2M31U75rqgu1wYjg78tyGZ9Xg 145
franken/data/base.py sha256=RuAzpSwlX6xV8rp5eT3yjQJP018MF_Dcae0o8iHkl8A 16778
franken/data/distributed_sampler.py sha256=lerkGoLs0RvuD6-s0TFa1oStW8RSW4N8JhEWYgG35PQ 1594
franken/data/fairchem.py sha256=zkbwM4TWmTbfQ4iTjDcCEi0mRHMNn9crz6NuwzWNoFk 3533
franken/data/mace.py sha256=Pxl7G4V8SzOyDjTykcgyrEJbh4p7HSPCWiOjhq5s-As 4136
franken/data/sevenn.py sha256=PcQO55iaISFgDv69L5Zyh7d6T_0of1Jy8Mj7P200SIs 5094
franken/datasets/__init__.py sha256=CM7U8kD6mzQzaU48Y9e38in19MGieMzUmh1cjE3r-2M 341
franken/datasets/registry.py sha256=gYbuT__H4ZtN-00cMf7sFqr4htSRH4mFYy_ypFn3Z1M 1845
franken/datasets/PtH2O/pth2o_dataset.py sha256=5x_RfAJCgemWCzb2IfmTdIg8FCckjwsbJpBCV6clJfk 2227
franken/datasets/TM23/tm23_dataset.py sha256=N3nC3r9_1_kMTXA5PGbyQCpOfJmWYF2JzzWW-tjI8Ss 2856
franken/datasets/test/long.xyz sha256=vqmzsOfBe1o56hzWvDClZPvRx7P9uv6gOGronAPd5yA 9948
franken/datasets/test/md.xyz sha256=uyTDVk4eKyMvtxDQct9f_fO7rtLN6TibNujehP3vnFc 561
franken/datasets/test/test.xyz sha256=qjIGMzKiq3WFQwB_WQcYXVuHBmxvW_z3VK-VWhNrtUk 526
franken/datasets/test/test_dataset.py sha256=F2hacokhxz1OM-8WTO0v2maFZwZiyl4HlA-UgHx5vh0 869
franken/datasets/test/train.xyz sha256=cIpkjVB2pngRApF38ZWglehHKAZ4bSYXhhrqjC1LzNQ 1055
franken/datasets/test/validation.xyz sha256=LnB_tQOqXCKgtE9HMF72Pe_bSkkHe-SIPn_mUzZV7CM 1055
franken/datasets/water/HH_digitizer.csv sha256=zq8AXaok_WOyEIedu0-EPJ79F68GCoV4spHOw8gD6Mc 1356
franken/datasets/water/OH_digitizer.csv sha256=I71zLPiav5UXjTCTbVUE1n-9QcgU2OAqauonXNqD3Ak 1327
franken/datasets/water/OO_digitizer.csv sha256=D0uNP7V2FkxmHl830ywSTgi3hbR0d5EczEMuOMf3o0w 1275
franken/datasets/water/exp_rdf.csv sha256=S-lRgCu2SPjagPEcLsRywuYYFgTP5aIL65EP1otXLOc 36118
franken/datasets/water/water_dataset.py sha256=R2y9UnaJS_wmy3KUNtEhb9yQ1z6GNj0WZ6Kze9hMmyM 6450
franken/mdgen/__init__.py sha256=b1JFV-OImvuTag-NW7qfBSj1vYEVCm2PntWBBgIwWi8 195
franken/mdgen/__main__.py sha256=-SJ9QlUpYK1thWzIZyzyz1eFeYZHS2JWVpGXsnKsKLc 105
franken/mdgen/asemd.py sha256=NKTilSx9LrXa-SOAtCzo3d1nCXtnQxMC3mK-DXh4otQ 11338
franken/mdgen/diffusion_coeff.py sha256=CoMasUG3FZiIHpVTikJVJ6AMfBHX8g9PghyDNk0W2hY 2868
franken/mdgen/mdgen_utils.py sha256=RUFjiH0QgOIvWHAK-DjKIixvcfd5puDQeK7w7GcpDPQ 3905
franken/mdgen/postprocess.py sha256=9c_jJXXz2i77SR5GXoOb9kWFXbOYeVCyoA_KSxcHvQg 11001
franken/mdgen/rdf_utils.py sha256=bNqSSHiHFMtnTlYkGX51QDVs5Erwl-aXB7JrFg5j0eA 9615
franken/mdgen/tm23_utils.py sha256=TI_LwCI3Ar3PhBPNS1n9sxYuPxztnZJPZbbJtZYZeXg 1444
franken/mdgen/configs/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
franken/mdgen/configs/md_postprocess.yaml sha256=dfbVYrdZmReyCUSVkfp4_iHsfS2n_UnFaB0Yrq2bQvk 541
franken/mdgen/configs/md_run.yaml sha256=HkIOjUeFonj4TZ09NmP3kN6HLxF0Be0aU3tRSzxYql0 336
franken/mdgen/configs/calculator/fairchem.yaml sha256=g7tMCNAcqI5_awL9eZKfARLt0gcXqfyB2mkmsi27xeQ 187
franken/mdgen/configs/calculator/franken.yaml sha256=bdBNdhTac7zscICd9TO2sQHze4G5E4Bc9GZPrCppcns 249
franken/mdgen/configs/calculator/mace.yaml sha256=GwHmLoPmVtpayjGD45slzx7hD1wAJ6aNlERv29NW7E8 318
franken/mdgen/configs/calculator/macel0_zeroshot.yaml sha256=tdzfn7znaEuXWCmOmtKeTZfmjqvlcFU7B86p_SGbO7w 347
franken/mdgen/configs/hydra/job_logging/custom.yaml sha256=YVI_c6hLdyNYxALTg_ZSOyuX2b4FkIbLbKQxlVpcmU8 92
franken/mdgen/configs/integrator/langevin.yaml sha256=DUxq9Qdy-nyZkoEiFPuSPpc1Helrq5QnHfSHwfwMKf8 161
franken/mdgen/configs/integrator/npt.yaml sha256=e8UcZ4zxaC3BpOSvyrBvlsaAd5J0zEihMXtMrVjZHyc 117
franken/mdgen/configs/md/TM23.yaml sha256=DcPk8q3t_2LAouuk3AfuygjU7A6tJ075Wid40xbg1YI 835
franken/mdgen/configs/md/franken-exp.yaml sha256=UJ4Am6_CRlFk9U5LF6lQaFaxbppMpLnfDhVssAvZDVI 929
franken/mdgen/configs/md/mace-exp.yaml sha256=dHj4Wug6Ss9XgcSfTr6HyqW560aTDv2XtfAs34OUQ6g 955
franken/mdgen/configs/md/water-diffusion.yaml sha256=tRXOs-JFN2p04ILEGZc5q-Ap6qEbAmKKl01bjuOmEkg 526
franken/mdgen/configs/md/water-liquid-mace.yaml sha256=JAHY6iLb5RYsBKGY2zg-9pGDBRNjRALeZEXDr1Elx6Q 792
franken/mdgen/configs/md/water-liquid.yaml sha256=EDaZdVYKfI0TY1bCqTzqSs64qYG6AOmuehlxyohY-mI 895
franken/mdgen/configs/paths/default.yaml sha256=ZkdU8ESkFMIHd4_58Um-JFTfpb4VW3YrcaCwnqYFgPc 92
franken/mdgen/configs/postprocess/TM23.yaml sha256=j4PuMatUzYJznQZ-2KFcZ7dGsaoB7qq-TFMKbujSogY 373
franken/mdgen/configs/postprocess/water.yaml sha256=g9ocUPAQFsFdPSFP4y7JQNPdn9vDZXmPyk0yfWbpMj4 253
franken/mdgen/configs/postprocess/water_diffusion.yaml sha256=EQkuX1gKGrtsvKFhdMU0aRc_qrktWg4Eiw_Dt8T6iKQ 377
franken/mdgen/configs/postprocess/metrics/rdf.yaml sha256=_w7zRfLGMYwH5i_eN93-_O3avlxqzzOAK5NSowj8G0A 258
franken/mdgen/configs/postprocess/metrics/water_diffusion_coeff.yaml sha256=MN_sHhSJ3zAFaUE-eK8iXM0C2A7t2c2LeFpTk9bwCk4 136
franken/metrics/__init__.py sha256=qC9A39BW6bTwH2JEkczad4m5fyBfiXry0f0e22r5f8A 519
franken/metrics/base.py sha256=holgvCl4afc7Fm_rEGVqyLBpnskfgUJmaD-MwxWlfXI 1710
franken/metrics/functions.py sha256=g91voXXEU5AG9kDVkmqMXMKxiF4mkSo-zH1SqvRB1nE 3655
franken/metrics/registry.py sha256=4aqaVKMjqsr4rEr2YXZJIXLLe3bJzp_WsQ_4bloNF2Q 966
franken/rf/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
franken/rf/atomic_energies.py sha256=tVJrBEUwk2Zqyx1DFqGeVtaz6YamR1q3Lk7_hLzzimA 2550
franken/rf/heads.py sha256=5NlGnfoy_FLUuAFQkHPqh_SrNysn4_2Bv8W8fT4XcX0 21582
franken/rf/model.py sha256=4eRlyD5zVnVkU2P7tFFckKpL-o-hDJ1-YizTiJWwinA 19038
franken/rf/scaler.py sha256=ulnlgB4I4Xr3p38L623TeUFmIBlFTXExaRqTUErQl_M 7839
franken/trainers/__init__.py sha256=Vyrle5rbbL7FdxWKc62Vsp53KfjHPhaLqTj-OvqfK68 175
franken/trainers/base.py sha256=BwMHbjhvyx8oLW0hWg8K6V9GnL8fXZER0GXw4R-yy5c 5074
franken/trainers/log_utils.py sha256=AawY8BZgek1jo4BTkWtFH5mvd2pq4NRyA23KQJIrwQE 10280
franken/trainers/rf_cuda_lowmem.py sha256=ljfRq_JQhrIYuVOBSkDP9sp1bA6asRYHo9sFJ1whFPI 14433
franken/utils/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
franken/utils/distributed.py sha256=uSf4Eiql4-TPguwJHP4w9ppl0XjWeg6GnaThvlvNDCE 4519
franken/utils/file_utils.py sha256=pIwa24WHBb8LLfb2PBOajTzO_H_dMBlVKEJFkhOGgCc 2198
franken/utils/hostlist.py sha256=q07slroG_C74tMUV_UnghI7kL2YicRxUDoAu122cs5c 13607
franken/utils/jac.py sha256=EFjeEM8lhLuptEKb20L-uCq3oUvzApM8CAQzfkm7T14 6101
franken/utils/misc.py sha256=VznmKSAFoiAGNbdN8WnWYpSjedkeAuaKrINWfnQc0ik 12779
franken/utils/linalg/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
franken/utils/linalg/cov.py sha256=0C5Y4T0eEe4NVsA9ygpLWpDgN-n8DKQPaXpIu5rEBPk 8270
franken/utils/linalg/psdsolve.py sha256=BaOVYVcb4ua93QMFOsa3VLE3kdBItd1DOwqyPocp9kI 4769
franken/utils/linalg/tri.py sha256=p1yalEiAsUV3IFV8LQRuBhwof1fZNyM1Bs1XPaUQEkI 8076
franken-0.3.3.dist-info/METADATA sha256=r9Qzo-gQqBi4HnrqcJTTpow4g-1FYhuw2UKMtrK6Mz8 4512
franken-0.3.3.dist-info/WHEEL sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ 87
franken-0.3.3.dist-info/entry_points.txt sha256=FUQa8xjKrtoIma-RdQkh5USso6Ya0rbKUD90edJXIBw 333
franken-0.3.3.dist-info/licenses/LICENSE.txt sha256=y-Kb9VjSKeXTpbGlOqMOrkR_Om34mk3QsMn6SPjd724 1059
franken-0.3.3.dist-info/RECORD

entry_points.txt

franken.asemd = franken.mdgen.asemd:md_cli_entry_point
franken.autotune = franken.autotune.script:cli_entry_point
franken.backbones = franken.backbones.cli:main
franken.create_lammps_model = franken.calculators.lammps:create_lammps_model_cli
franken.mdgen.postprocess = franken.mdgen.postprocess:pp_cli_entry_point