Wheelodex » Projects » apax

apax

View on PyPI — Reverse Dependencies (0)

0.4.0 apax-0.4.0-py3-none-any.whl

Wheel Details

Project: apax
Version: 0.4.0
Filename: apax-0.4.0-py3-none-any.whl
Download: [link]
Size: 124490
MD5: ae677c7046e0b79f366839e32e777419
SHA256: 05b2c78cadb65aedb991cd2b97b2a182d4c4e4ff2fa4d37ced81ddb5205b0e2f
Uploaded: 2024-04-10 21:06:53 +0000

dist-info

METADATA · WHEEL · RECORD · entry_points.txt

METADATA

Metadata-Version: 2.1
Name: apax
Version: 0.4.0
Summary: Atomistic Learned Potential Package in JAX
Author: Moritz René Schäfer
Author-Email: schaefer[at]theochem.uni-stuttgart.de
Project-Url: Documentation, https://apax.readthedocs.io/en/latest/
License: MIT
Keywords: machine-learning,interatomic potentials,molecular-dynamics
Classifier: License :: OSI Approved :: MIT License
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Requires-Python: >=3.10,<3.12
Requires-Dist: ase (<4.0.0,>=3.22.1)
Requires-Dist: chex (<0.2.0,>=0.1.9)
Requires-Dist: clu (<0.0.8,>=0.0.7)
Requires-Dist: einops (<0.7.0,>=0.6.0)
Requires-Dist: jax (<0.5.0,>=0.4.25)
Requires-Dist: jaxtyping (<0.3.0,>=0.2.8)
Requires-Dist: lazy-loader (<0.3,>=0.2)
Requires-Dist: matscipy (<0.9.0,>=0.8.0)
Requires-Dist: mlflow[mlflow] (<3.0.0,>=2.11.3)
Requires-Dist: numpy (<2.0,>=1.23)
Requires-Dist: optax (<0.2.0,>=0.1.3)
Requires-Dist: pydantic (<3.0.0,>=2.3.0)
Requires-Dist: tensorflow-cpu (<3.0.0,>=2.12.0)
Requires-Dist: typer (<0.8.0,>=0.7.0)
Requires-Dist: znh5md (<0.2.0,>=0.1.7)
Requires-Dist: zntrack (<0.8.0,>=0.7.2); extra == "zntrack"
Provides-Extra: zntrack
Description-Content-Type: text/markdown
[Description omitted; length: 4338 characters]

WHEEL

Wheel-Version: 1.0
Generator: poetry-core 1.7.0
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
apax/__init__.py sha256=tXeRstB01WwQTGSiBER6j1XYwaOoaQIBScNg0g1ya0Y 214
apax/bal/__init__.py sha256=r3jKWeQlE0cbH4xBsuLz_qNdFdZnF_V4CqwxXbFuRz0 74
apax/bal/api.py sha256=lWt_ykl57e_SKts9b_ih7IBDU1c__zdAaJsgz1NzwPY 5053
apax/bal/feature_maps.py sha256=tf-roM8G-2UzzlicxROvnKvAKK8TsGpnGj9VXNpIfVw 2852
apax/bal/kernel.py sha256=P1FAw9bYrOJARFKblPWp24t4f74Hgt0TTdVO4mlhB28 770
apax/bal/selection.py sha256=MGZq29-phzfDJK8pT-_nbl4stc7-EDE1OYkAqGMPVe8 1267
apax/bal/transforms.py sha256=hLsb1i_A6Oyl5_c4FcY6h4xAINWuBiu7WOcAvkRxCkw 1154
apax/cli/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
apax/cli/apax_app.py sha256=tTIIhScrFndXhg-tkcnJI-Ry2AtQstqzadyZzLh4L8w 7653
apax/cli/templates/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
apax/cli/templates/md_config_minimal.yaml sha256=3o9dW5SD0YQaRwY9sPotr66WwVJpWIpalltbuWi-O8s 481
apax/cli/templates/train_config_full.yaml sha256=dYfyhbYWL_8jbfrFnPl3snxDQ-6vGbcuTy8NJyRLhq8 1565
apax/cli/templates/train_config_minimal.yaml sha256=toH8ERX00ZnQcIPywkLDwNKg2_Z_jFVN94zJwXF8M1E 319
apax/config/__init__.py sha256=OHb6Xou2v6u42swTgjRfzej4CIlRg4OmgOIQXUiRjKA 97
apax/config/__init__.pyi sha256=RHK2Lhw08k1aQYh0mZuagThQtJlMjM7RIsaIitMj8V8 402
apax/config/common.py sha256=YoBAf3O0j0ZPX1If79dqzEmkCZ782zzqH5JPEmC07Es 1488
apax/config/md_config.py sha256=lApskQ_0ttVFG-beP3DMs2sDAX2XotADcUUVFlejrY4 4951
apax/config/train_config.py sha256=dM8CL8U-CR88n2itkQ6Hdf94Vd9AVBbku0NmgPQV5Bs 14279
apax/data/__init__.py sha256=OHb6Xou2v6u42swTgjRfzej4CIlRg4OmgOIQXUiRjKA 97
apax/data/__init__.pyi sha256=lRS0L8a6VnHRX0aTJO95hS_vvbYpyEL1jLxsC3VL5FU 121
apax/data/initialization.py sha256=V0LN707QSvNZKrB-335hmwPGp182gHeUEvjxGPce8V0 1479
apax/data/input_pipeline.py sha256=KmO_aZZ-cpBdaMlpMf6ZzfT-K270oVtM80ry4HeITdk 11189
apax/data/preprocessing.py sha256=NwgBHs-fp3En6EPH9Bu6hhTU9Da-nQuWB34JFpKV4cs 3239
apax/data/statistics.py sha256=brE7xoKlgNo7nS1SCu6EBmmGqXKag66dq9q27yGbsV8 5512
apax/layers/__init__.py sha256=r1lTlRAz93ACzicKoP-UA0wTGMgLYCQrINT_1Imic6o 133
apax/layers/activation.py sha256=hR4LWRudrfCpzqFPcgNuz_VRXuOVPgeKRGRleZzE1c0 89
apax/layers/descriptor/__init__.py sha256=XE95Xsj3bPxcDtyiugRBy__xQYCxoxwo7c81CAQP1ac 127
apax/layers/descriptor/basis_functions.py sha256=sWQtUcEaEe32GPmLx1tOsX3tWRCehEDvnLjTIRiMvcs 3318
apax/layers/descriptor/gaussian_moment_descriptor.py sha256=5fv7qGEqO5ijToj5TlYp_zZD66cNpfMK_vwHSBG6L1Q 3998
apax/layers/descriptor/moments.py sha256=gsbeZ-6QATvhMWQaJLEtZwy24SKfrF5M_z9-mcK2xDg 1127
apax/layers/descriptor/triangular_indices.py sha256=h9OBhrFVOov7BgynahGJDV4NUG-SOTtGyblBeayLIxo 1326
apax/layers/empirical.py sha256=a5vCXhtYA2wf7s49zcwZrKxIrwerzuyaWNvgkD5uyZ0 2635
apax/layers/initializers.py sha256=rzTtN-wfxPdjTVynJulAfjOoVS9uFGH8SoMbqot7fLo 676
apax/layers/masking.py sha256=35wu2zSCA7EohHwgEneifeU2lRqeW1vYkL4pwvow0aU 357
apax/layers/ntk_linear.py sha256=Ze135mtQ-nHqCRhH9lIv70pACr3pDM3dJmjMg93fS4k 1116
apax/layers/properties.py sha256=W7Y9FnoPS7tbVDcfVF-rT1mx1v2UTMA9ol5BT3vZQdc 1145
apax/layers/readout.py sha256=6yvAR0U3eKAKlDSfNa62TeM7AIg6053KoiqeFyPsbuk 915
apax/layers/scaling.py sha256=Uh3DdaQtAa7FTZKWvATQyjIdJoNPVFabR5zW6FKBbJ4 1326
apax/md/__init__.py sha256=OHb6Xou2v6u42swTgjRfzej4CIlRg4OmgOIQXUiRjKA 97
apax/md/__init__.pyi sha256=xHc2KhdDoW7M-9pLichBaKlFhwM0ed8ur8ve0cjchqg 99
apax/md/ase_calc.py sha256=Itv6k7gkpZ7PY26wkLuWa_eYGgvBplKTauJZ5Rjls1U 13217
apax/md/function_transformations.py sha256=EKbAnl0er-iKqMNdf-1B9cglqz-nZDEw8LPfjh6PHZg 2807
apax/md/io.py sha256=BxN0aRlC_A6BTsKAMXBVgKKKyP4BdUvrffs50iE_JpQ 3788
apax/md/md_checkpoint.py sha256=6RwugPb9a6LMPu3VakCfp3qKo_B0hjiYX063CvoRVqA 589
apax/md/nvt.py sha256=fPK0spokg9TvOoP4NGRt4kHfZO3by4o4m1UH5NSA7S8 13504
apax/md/sim_utils.py sha256=E4IrCVLunmPwwlJg2EX9B8Pf7OAdq113iC4U4x6bqko 1128
apax/model/__init__.py sha256=OHb6Xou2v6u42swTgjRfzej4CIlRg4OmgOIQXUiRjKA 97
apax/model/__init__.pyi sha256=ykOti2sqpg33kfs3EOnZ6IYjA7vyx1iPSmtd0xHgYRc 189
apax/model/builder.py sha256=W-z03l-58Kgnj107WvobqrmDoUZsBN78m0aoUuq6N4U 3818
apax/model/gmnn.py sha256=YW6EpkwgCm8--woHwvDwKXW115n6Rvl7hlBRyrf8IV4 5357
apax/nodes/__init__.py sha256=dzDlGaqDHqOuYH7AQmAtI29owAlkh91AC7JlbAfbi7g 113
apax/nodes/md.py sha256=-cHjxyNgGYpa6srKFeEVtlOhfjyv1Jm-BXRi665oZH0 2405
apax/nodes/model.py sha256=_IWR4nWqlsm4iyYsppO6q1BAoZ-e-7--nRUYjGlb2Hw 4622
apax/nodes/utils.py sha256=i17FBf24rxoG7YPlYcYxlBDBod2Opbj6oYaxnhKMG-o 777
apax/optimizer/__init__.py sha256=VHDzmly0IIDdgHJ-vY5pXzF3tRo6rIhgB53C3Lr2-eA 72
apax/optimizer/get_optimizer.py sha256=ZXIIvJfAMGyDHmuW3RbHnP9IWk2vnbUqb5xMjfQbMFw 2848
apax/train/__init__.py sha256=po2JhqqjgcFWatxYLKviMNvmKOghBoisykaUbfORh30 163
apax/train/__init__.pyi sha256=UOySKCHWw1-JBecKF4T1HwypRwolHWVBRD3TftZUfvs 178
apax/train/callbacks.py sha256=VHfwWXLjrk7GVoHcRDpi1zwoknA7CN3UrbY1BrnyGPI 3645
apax/train/checkpoints.py sha256=87w3_m5jiPh3gjiyvUqSk1XocLP9E8O7Kujb19iD3xs 6467
apax/train/eval.py sha256=fMCPEC68IPUgyaCB9ZEXyKXZS-uSKqeQ4SkFH6353Oc 5736
apax/train/loss.py sha256=9yYzvXsXdJFkaktg9Avmz3esDwczIXHKj0Z66s1MKEg 4564
apax/train/metrics.py sha256=wuQDD-DG4jh9g59736dDuNNYzyHY_WoLbf1JMu_Yr1U 2775
apax/train/run.py sha256=zW32cp1ebiNtjgflV_bKOV4jUauilx2dGz9E9pkNlYo 6290
apax/train/trainer.py sha256=l1nD8kd_5BeOrbMFROHPJcqYHowQrjSiPu20HtRVx3Q 10236
apax/transfer_learning/__init__.py sha256=STMdzPlGpJXZ4twf1ftCqtXckLFvNsaQaaXHIOvKGQY 178
apax/transfer_learning/parameter_transfer.py sha256=TkbDF9pK-Wei_O-Jxn6zWLxiV_Co-7SrRi6e_tz_SJA 1135
apax/utils/__init__.py sha256=OHb6Xou2v6u42swTgjRfzej4CIlRg4OmgOIQXUiRjKA 97
apax/utils/__init__.pyi sha256=0xSnnn3RKiwssSDCzaLHuA4p30oYmjbMxPB4UtQfOnI 145
apax/utils/convert.py sha256=b2XcCFSbKRLhm5bJ8kDTdJLnOrS8SxoI54bv14XBMK0 4283
apax/utils/data.py sha256=flvCgx17MWGuC77IrgCy2xsQyWnZZxwu2gkbkbZ3t00 3093
apax/utils/datasets.py sha256=LD3jMRDYk_AoYia3bttGR88Mz3AUKxGWk-8jPVqGoKE 3088
apax/utils/helpers.py sha256=YRYAJ0DpxQ0aBuT56OQ9cJwbP3Wpnnea4fF54pIqwhA 1464
apax/utils/jax_md_reduced/README.md sha256=-6W_j_cWOC94ycwgFraEVu4Z3kRH-oh4PdKyok_sL-U 726
apax/utils/jax_md_reduced/__init__.py sha256=Vw2FXPUCHfF9rVObja7xuWAFjL5mqgsGZHAgd1nOCoo 222
apax/utils/jax_md_reduced/dataclasses.py sha256=ZqQTvAGC0kBEsZevFnqkeaGAn8r-WREeyVKGPf6nj94 2548
apax/utils/jax_md_reduced/partition.py sha256=WrcQHAkh64Dc0RMI-RvKRC3GzDKoiCBtRT55hi1NbZw 31392
apax/utils/jax_md_reduced/quantity.py sha256=iva1lQl_hpclCJDoRqeyR4qOvZLM2Og1Mwf4j5qh004 26600
apax/utils/jax_md_reduced/simulate.py sha256=QRhNB8G6SsqqlLKOMZ5CeKr1ssbXLkurN4jeO5R8Hrw 62016
apax/utils/jax_md_reduced/smap.py sha256=GmvOgmR_fUWfNzPBuxrm5zD4WdO236WlohbUKKd-V6E 37596
apax/utils/jax_md_reduced/space.py sha256=DE0diCMfYNdJEo2oT-DVwbLa_sXfIfScNDJNDyOZa50 11931
apax/utils/jax_md_reduced/util.py sha256=OPJU8ZVN5g6KnHh5Jj9KlG_BgZf3NrYiXo6zghsNjsY 2864
apax/utils/math.py sha256=u_7jzLV0xokIy6pKVS_tSBoShhCo_9WW4HYhpeE80aU 702
apax/utils/random.py sha256=ShL0hhjgDNhd1hTKWSZHT7HtI6_CztxRXFt9Q8-77UU 170
apax-0.4.0.dist-info/LICENSE sha256=D1th3_Mod1zsxijBAneLP08ClqxdbQWMZwmm0Kko4Co 1089
apax-0.4.0.dist-info/METADATA sha256=GwnzI-A1UvZW6Sl8MAVgUlWQuhGwmAGSeropyKBy2k8 5642
apax-0.4.0.dist-info/WHEEL sha256=d2fvjOD7sXsVzChCqf0Ty0JbHKBaLYwDbGQDwQTnJ50 88
apax-0.4.0.dist-info/entry_points.txt sha256=egIp1X0oJDgpml9cIeC2nmnwi_PAozR198TCswSc3g8 46
apax-0.4.0.dist-info/RECORD — —

entry_points.txt

[console_scripts]
apax = apax.cli.apax_app:app