Wheelodex » Projects » apax

apax

View on PyPI — Reverse Dependencies (0)

0.7.0 apax-0.7.0-py3-none-any.whl

Wheel Details

Project: apax
Version: 0.7.0
Filename: apax-0.7.0-py3-none-any.whl
Download: [link]
Size: 143755
MD5: b19f8e3516f9ea6ada954b10c2549067
SHA256: bdf71b2ecf43246030bb1e7d2f7050c8ed62414afbe85c45bb0af8a133d119ec
Uploaded: 2024-09-17 10:55:44 +0000

dist-info

METADATA · WHEEL · RECORD · entry_points.txt

METADATA

Metadata-Version: 2.1
Name: apax
Version: 0.7.0
Summary: Atomistic Learned Potential Package in JAX
Author: Moritz René Schäfer
Author-Email: schaefer[at]theochem.uni-stuttgart.de
Home-Page: https://github.com/apax-hub/apax
Project-Url: Documentation, https://apax.readthedocs.io/en/latest/
Project-Url: Repository, https://github.com/apax-hub/apax
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: e3x (<2.0.0,>=1.0.2)
Requires-Dist: einops (<0.7.0,>=0.6.0)
Requires-Dist: flax (==0.8.4)
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 (<3,>=2); extra == "mlflow"
Requires-Dist: numpy (<2.0,>=1.23)
Requires-Dist: optax (<0.2.0,>=0.1.3)
Requires-Dist: orbax-checkpoint (==0.5.16)
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: uncertainty-toolbox (<0.2.0,>=0.1.1)
Requires-Dist: znh5md (<0.4,>=0.3)
Requires-Dist: zntrack (<0.8.0,>=0.7.3); extra == "zntrack"
Provides-Extra: mlflow
Provides-Extra: zntrack
Description-Content-Type: text/markdown
[Description omitted; length: 3986 characters]

WHEEL

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

RECORD

Path Digest Size
apax/__init__.py sha256=lXXQkGjmQ7-ODD9RXdgkHlT4Rlgd_lJs7whltGJmswU 289
apax/bal/__init__.py sha256=r3jKWeQlE0cbH4xBsuLz_qNdFdZnF_V4CqwxXbFuRz0 74
apax/bal/api.py sha256=07IDL-EKbHRLYR4VsUErLLMTcPvIxrDKslurEPJyb-8 5051
apax/bal/feature_maps.py sha256=5AOwlh9iXrDE3pbAqskBjY9QN2AZ9XxKlypIfZWNikU 4953
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/calibration.py sha256=Xg1w3z3cGj1tI-pOhwKNA-3zYQPKK2fN1uOfyJ3rTDo 2208
apax/cli/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
apax/cli/apax_app.py sha256=gix7oOcMF1X5Wu9yZU8iIxv7gxSmiVst1hjJXFCcAf8 9051
apax/cli/templates/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
apax/cli/templates/md_config_minimal.yaml sha256=iehw6SOXm9qAjrTVxwsJzXjAg2Nb8wcSGAkuRXVQI4s 517
apax/cli/templates/train_config_full.yaml sha256=1cMwo-IxH_FGMLSOlvQ1_-baWSXeul8rVVsUU4R_-v4 1848
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/lr_config.py sha256=1TnIZ4FyfZJA0BzsjVqoWubokuiAzrOI4IPNkw5_zcM 1153
apax/config/md_config.py sha256=UCCJXOHX_OQgKOKe8hOOzFrk0_OcNlXxS0fBtW5T8vw 8678
apax/config/model_config.py sha256=iTD0BMM3lqobZLkk-Yn0et703gunohOxppoRMDt8Lkc 7713
apax/config/train_config.py sha256=fwVwKQvMQb0VrH6JlXtywhtVhoHyN_f82k3JlixblQ4 14469
apax/data/__init__.py sha256=OHb6Xou2v6u42swTgjRfzej4CIlRg4OmgOIQXUiRjKA 97
apax/data/__init__.pyi sha256=lRS0L8a6VnHRX0aTJO95hS_vvbYpyEL1jLxsC3VL5FU 121
apax/data/initialization.py sha256=ze2uu40jrAfbQcBxkNUGR73woIePfY5NLQMd5EHHwRU 1668
apax/data/input_pipeline.py sha256=l4WYcA4JJw4cPSTzP0GBQxywHOPRHUj9YQJKq7NV_RE 18325
apax/data/preprocessing.py sha256=NwgBHs-fp3En6EPH9Bu6hhTU9Da-nQuWB34JFpKV4cs 3239
apax/data/statistics.py sha256=t9QUqk6Y05WTF1t_0iqBPjxKkII3OHrR353wvJsEgXk 6014
apax/layers/__init__.py sha256=r1lTlRAz93ACzicKoP-UA0wTGMgLYCQrINT_1Imic6o 133
apax/layers/activation.py sha256=hR4LWRudrfCpzqFPcgNuz_VRXuOVPgeKRGRleZzE1c0 89
apax/layers/descriptor/__init__.py sha256=tdytZMSnnr5GFocd3f_MBx-LJpvvAX7gnNSNjT8mHrU 379
apax/layers/descriptor/basis_functions.py sha256=rx9_oGTFhfBKk92GLQ4FmslfhfzF1OsE2BZzbNoARO0 4232
apax/layers/descriptor/equiv_mp.py sha256=xRE4OCpNALbPKqebGoWkuXhrPmKZk1Hv49fvFW73LI4 2992
apax/layers/descriptor/gaussian_moment_descriptor.py sha256=bGr2Y9ZRd5LamT736CdKqANYW86GyFK-WSzPOwebGK0 3976
apax/layers/descriptor/moments.py sha256=gsbeZ-6QATvhMWQaJLEtZwy24SKfrF5M_z9-mcK2xDg 1127
apax/layers/descriptor/so3krates.py sha256=1skEI3ohErzVXcpg1A1E4c3g9nVgub859-ES0I8GUD8 3434
apax/layers/descriptor/triangular_indices.py sha256=h9OBhrFVOov7BgynahGJDV4NUG-SOTtGyblBeayLIxo 1326
apax/layers/distances.py sha256=pDjVuqHZexDHY7LlKCSu3qppBOY36QvPZint7KQKPDM 2300
apax/layers/empirical.py sha256=xIEU4tAAUP-uFAumXpDwFuKwREhbJIFrs4ErltsZ8oI 2118
apax/layers/initializers.py sha256=rzTtN-wfxPdjTVynJulAfjOoVS9uFGH8SoMbqot7fLo 676
apax/layers/masking.py sha256=zwVDS6NmhyxT6NS_pYGfcHaquO3UVwYYMJuyTpKzS7s 428
apax/layers/ntk_linear.py sha256=7vRVQeeH8nDyx_8v458LmTk9XliOjhkBBUrm_j2wwbk 1588
apax/layers/properties.py sha256=W7Y9FnoPS7tbVDcfVF-rT1mx1v2UTMA9ol5BT3vZQdc 1145
apax/layers/readout.py sha256=DeuUh1Rz_X1W5A6nzibtO-cE2MTFGAGyDjI9jC_1KbA 1330
apax/layers/scaling.py sha256=Uh3DdaQtAa7FTZKWvATQyjIdJoNPVFabR5zW6FKBbJ4 1326
apax/md/__init__.py sha256=OHb6Xou2v6u42swTgjRfzej4CIlRg4OmgOIQXUiRjKA 97
apax/md/__init__.pyi sha256=n_JLjz9xPNvtC8HBdKuim3ErGFpMwJLOhUtuJZqQCs4 104
apax/md/ase_calc.py sha256=-5q3rFzPSSwWaaEI3dVHA5AOZyi3-U4hwu5YvIIX2S8 14987
apax/md/dynamics_checks.py sha256=XIPkWMOYy-X33fKKpNJbt4dPhT7y4IvYouM8qT3rx5Q 1496
apax/md/function_transformations.py sha256=qaATMsCzNP2goCdYZWK3HCfmcVgPCiNzm19e44qDZu4 4152
apax/md/io.py sha256=udUwi_wGsHcgqMD9Onm4_FviEvlTjdyGYIGbjSOk8gA 4136
apax/md/md_checkpoint.py sha256=6RwugPb9a6LMPu3VakCfp3qKo_B0hjiYX063CvoRVqA 589
apax/md/schedules.py sha256=CU_Al4DnTK_SKh_ptPfMrnq_3-oqK1Q9KYhZ_YxWfxU 1499
apax/md/sim_utils.py sha256=ZcB-SafPJJ8cNFPzTKfhhcbNPMltu3cuLFMHTTe9qDM 1155
apax/md/simulate.py sha256=P5l2z2qxVmtlh5Yag929a1F0Ur9uAUxk66IC9raZ1FU 17345
apax/nn/__init__.py sha256=OHb6Xou2v6u42swTgjRfzej4CIlRg4OmgOIQXUiRjKA 97
apax/nn/__init__.pyi sha256=VtYwn_oXwvASv3M8pMw6j9XWfwrZMuRbxMZXc1TWyYU 191
apax/nn/builder.py sha256=ZXbiKyB3t5ZzlFBxfWfeFwm1TzcL-JwCZcrLKuDgcBQ 7877
apax/nn/models.py sha256=-BiqmqQALvkwNVs1qT8ts_v7mBZ53xGR4ne1cT1G8k0 8872
apax/nodes/__init__.py sha256=ZDidR0a4mHlCTRx-6X-z4u88LCnqSzcFoxaoFHmWGoU 414
apax/nodes/analysis.py sha256=Y4gEFshx0r9s1BjJUIvwrmBivDMQflY_2Rhp7AKKjLY 905
apax/nodes/md.py sha256=yUvCSbcGEe-B2PmLAHSaagbDzL5CtJM9tWmWTgW30ao 2308
apax/nodes/model.py sha256=2mqDo8-xIfiOgUtz2WBstlVJFyNDfKaJKQVPjwdoc1s 10218
apax/nodes/selection.py sha256=L2DPmfQdEt2TEgq6h3V6Cda38M52gZqb-wJ_9yVSelM 3102
apax/nodes/utils.py sha256=i17FBf24rxoG7YPlYcYxlBDBod2Opbj6oYaxnhKMG-o 777
apax/optimizer/__init__.py sha256=VHDzmly0IIDdgHJ-vY5pXzF3tRo6rIhgB53C3Lr2-eA 72
apax/optimizer/get_optimizer.py sha256=ri_VicHC-vqwKMYIvIedyVrTHE7DKr-JaPFUiwlRBgI 4074
apax/train/__init__.py sha256=po2JhqqjgcFWatxYLKviMNvmKOghBoisykaUbfORh30 163
apax/train/__init__.pyi sha256=UOySKCHWw1-JBecKF4T1HwypRwolHWVBRD3TftZUfvs 178
apax/train/callbacks.py sha256=vhxNDwPaCWcm3cr4qBeEyvi_A-f9RkEsdOuN1zjaHt8 5261
apax/train/checkpoints.py sha256=MEINK4Ux8twgl7_skrj2nqKmJqtEpAL1kyE9ZfZgFpI 6469
apax/train/eval.py sha256=MnLOrZamBjJcrqikgPCtM7UMn26Vd9BBeUT51XvqxMU 5906
apax/train/loss.py sha256=TmuZeeObG5tsLOl3qStUbIx_P-NcUIAIaeiA6EsiNA4 6403
apax/train/metrics.py sha256=wuQDD-DG4jh9g59736dDuNNYzyHY_WoLbf1JMu_Yr1U 2775
apax/train/parameters.py sha256=CAQRL2bZfl2N2J_XpiTcdAaZ1fI60SjBezkAXfDhMZc 977
apax/train/run.py sha256=Wy-3Cf-eiP8Tc7vNmBSkcEn_PV4Rin4-ZVLd_SHUprU 6670
apax/train/trainer.py sha256=_ld_46O0EkYomLBIqDS2CPxUQpSQKABYEcNkIxINC2o 9875
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=8kM2QlsHmsx9hw5x2sDjXHOK4adftogtQ9CA-MYTcz0 4561
apax/utils/data.py sha256=xlii4wnKKzMZrn0AYlyuo25pzf3B0etipAn5fhP6_qI 2654
apax/utils/datasets.py sha256=LD3jMRDYk_AoYia3bttGR88Mz3AUKxGWk-8jPVqGoKE 3088
apax/utils/helpers.py sha256=AmsA3n15XMrYgst9DhiD1pmYUer12OlNxd8PuAVcV3s 1625
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=CsvOT0_OZP3gXCw369Ynh4gPBh8vhbH1q4Bryq4zWAM 31363
apax/utils/jax_md_reduced/quantity.py sha256=P422qcg0ZsHcRXI69427sCSX3GGhZIY8znpk1V6TT8E 26569
apax/utils/jax_md_reduced/simulate.py sha256=JG0tLYQtk8Zk9SmGNV5MIeUjbE7-nyyGR6xnRsDBY5c 62011
apax/utils/jax_md_reduced/smap.py sha256=i8dH4kUQI9KYWxwy3zvIxSf-fMIoPITD8cTW_bA2dX4 37443
apax/utils/jax_md_reduced/space.py sha256=SGaVtkEEOZRD8k5feaLjzOTZQHFd4N8W94R6IAaaOOk 11921
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.7.0.dist-info/LICENSE sha256=D1th3_Mod1zsxijBAneLP08ClqxdbQWMZwmm0Kko4Co 1089
apax-0.7.0.dist-info/METADATA sha256=XEs-US3HX2e6QaGhxLioMDS1IZweHukK7i-5j8pd0f8 5575
apax-0.7.0.dist-info/WHEEL sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg 88
apax-0.7.0.dist-info/entry_points.txt sha256=egIp1X0oJDgpml9cIeC2nmnwi_PAozR198TCswSc3g8 46
apax-0.7.0.dist-info/RECORD — —

entry_points.txt

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