Wheelodex » Projects » apax

apax

View on PyPI — Reverse Dependencies (0)

0.10.2 apax-0.10.2-py3-none-any.whl

Wheel Details

Project: apax
Version: 0.10.2
Filename: apax-0.10.2-py3-none-any.whl
Download: [link]
Size: 149048
MD5: 8231b52739cc6ea837774d905bc39b96
SHA256: 5b5bd51736dac19675795a3dff55d69889d7fdcd0d1fceedb871b271bf98fb8f
Uploaded: 2025-01-21 13:44:27 +0000

dist-info

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

METADATA

Metadata-Version: 2.2
Name: apax
Version: 0.10.2
Summary: Atomistic Learned Potential Package in JAX
Author-Email: Moritz René Schäfer <schaefer[at]theochem.uni-stuttgart.de>, Nico Segreto <segreto[at]theochem.uni-stuttgart.de>
Keywords: machine-learning,interatomic potentials,molecular-dynamics
Requires-Python: >=3.10
Requires-Dist: ase (>=3.24.0)
Requires-Dist: clu (>=0.0.12)
Requires-Dist: e3x (>=1.0.2)
Requires-Dist: einops (>=0.8.0)
Requires-Dist: flax (>=0.10.2)
Requires-Dist: jax (>=0.4.38)
Requires-Dist: lazy-loader (>=0.4)
Requires-Dist: numpy (>=1.23)
Requires-Dist: optax (>=0.2.4)
Requires-Dist: orbax-checkpoint (>=0.11.0)
Requires-Dist: pydantic (>=2.10.5)
Requires-Dist: tensorflow (>=2.12.0)
Requires-Dist: typer (>=0.13.0)
Requires-Dist: uncertainty-toolbox (>=0.1.1)
Requires-Dist: vesin (>=0.3.0)
Requires-Dist: znh5md (>=0.4.2)
Requires-Dist: jax[cuda]; extra == "cuda"
Requires-Dist: mlflow (>=2.19.0); extra == "mlflow"
Requires-Dist: ipsuite (>=0.2.3); extra == "ipsuite"
Requires-Dist: sphinx (>=7.2.6); extra == "docs"
Requires-Dist: sphinx-rtd-theme (>=1.3.0); extra == "docs"
Requires-Dist: myst-parser (>=2.0.0); extra == "docs"
Requires-Dist: sphinx-copybutton (>=0.5.2); extra == "docs"
Requires-Dist: furo (>=2023.9.10); extra == "docs"
Requires-Dist: nbsphinx (>=0.9.3); extra == "docs"
Requires-Dist: nbsphinx-link (>=1.3.0); extra == "docs"
Requires-Dist: ipykernel (>=6.29.3); extra == "docs"
Requires-Dist: pandoc (>=2.3); extra == "docs"
Provides-Extra: cuda
Provides-Extra: mlflow
Provides-Extra: ipsuite
Provides-Extra: docs
Description-Content-Type: text/markdown
[Description omitted; length: 3802 characters]

WHEEL

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

RECORD

Path Digest Size
apax/__init__.py sha256=lXXQkGjmQ7-ODD9RXdgkHlT4Rlgd_lJs7whltGJmswU 289
apax/calibration.py sha256=Xg1w3z3cGj1tI-pOhwKNA-3zYQPKK2fN1uOfyJ3rTDo 2208
apax/bal/__init__.py sha256=r3jKWeQlE0cbH4xBsuLz_qNdFdZnF_V4CqwxXbFuRz0 74
apax/bal/api.py sha256=6Hv32TLE0oHJGB46_218J66b_1bP2d0KOcH069ihE9E 5284
apax/bal/feature_maps.py sha256=73eNU8jYOpshfaATfTA6gVAmsaYHNCVFoRbsRJUE7h8 7335
apax/bal/kernel.py sha256=P1FAw9bYrOJARFKblPWp24t4f74Hgt0TTdVO4mlhB28 770
apax/bal/selection.py sha256=JSZ0lSoh6WnyRzJDC7PrONbaLRl8vnz3fEjDfbJk3L4 1619
apax/bal/transforms.py sha256=hLsb1i_A6Oyl5_c4FcY6h4xAINWuBiu7WOcAvkRxCkw 1154
apax/cli/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
apax/cli/apax_app.py sha256=XoOq88jzdBVOw8yFYTLqGT_g1sAXhS9UWUUvPlNznRg 9053
apax/cli/templates/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
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=n8yZlbqb2jstkuT40AohTyADA9A6rzpFU6pd5VjCXvs 1155
apax/config/md_config.py sha256=s0LB5fDp7UiyyWClUC-kjDpAJGV923VdJs5Ovs-SInY 8952
apax/config/model_config.py sha256=kV7FYXJrWDWyGeF__mcNa-lfQ2Ja5s-k2-LYIYxdq6U 8325
apax/config/train_config.py sha256=81toqd0xAB-TkNiQ8d-pU2M8MecbYpRTQLv5uUB3qBI 15074
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=_Y1wDFFexmilmPdqhBI9Cnu2xvRxSBU4WbliGhNyW5s 21850
apax/data/preprocessing.py sha256=ZKKpkutmTK9GJ2rrIdF_msjQ9_E66IYEjQ2VG5be6Lk 3231
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/distances.py sha256=pDjVuqHZexDHY7LlKCSu3qppBOY36QvPZint7KQKPDM 2300
apax/layers/empirical.py sha256=sjrGfQynFe4ZFG2V1rRGiCzuexmqBOfORCXEevS2cg0 5086
apax/layers/initializers.py sha256=rzTtN-wfxPdjTVynJulAfjOoVS9uFGH8SoMbqot7fLo 676
apax/layers/masking.py sha256=JMT45gWt6mACLh7LH4SMUYxPC73m8KNSn7-E0n2Vtqg 455
apax/layers/ntk_linear.py sha256=fVnplIYCrMrDOBR9R1ZwMBEIDjiR92xzfw2Zu91g-nk 1605
apax/layers/properties.py sha256=sukZsWxmRWAeREiRFNj6PAB73Oq3H3U7rNs3h8ZlGTo 3901
apax/layers/readout.py sha256=BVL6Mu0eJRtKRPMAGedV2_W3N1QA4fXsbQiItMZI5ts 1457
apax/layers/scaling.py sha256=Vtl0cVztzZuiQqbP7eQ4dgRabAMnS3JWwCaVZ9lfamo 1433
apax/layers/descriptor/__init__.py sha256=tdytZMSnnr5GFocd3f_MBx-LJpvvAX7gnNSNjT8mHrU 379
apax/layers/descriptor/basis_functions.py sha256=C0xpB54LzEOqblHCJjCnzPs_UM4VphGGaKeeDNJzYrw 4384
apax/layers/descriptor/equiv_mp.py sha256=paUdRXEe7grPA_YToDI2D9wvUiVKZpVplUczgZ1vrQo 3057
apax/layers/descriptor/gaussian_moment_descriptor.py sha256=H9M7OBfmjAUF4H1m20X8KsxKOw6oTeXBvQBjRnV3Kto 3678
apax/layers/descriptor/moments.py sha256=gsbeZ-6QATvhMWQaJLEtZwy24SKfrF5M_z9-mcK2xDg 1127
apax/layers/descriptor/so3krates.py sha256=aPa84G0xvWsLvC_l2x0mq5uCZWp6UjYTKvCpRuon-OI 3452
apax/layers/descriptor/triangular_indices.py sha256=h9OBhrFVOov7BgynahGJDV4NUG-SOTtGyblBeayLIxo 1326
apax/md/__init__.py sha256=OHb6Xou2v6u42swTgjRfzej4CIlRg4OmgOIQXUiRjKA 97
apax/md/__init__.pyi sha256=n_JLjz9xPNvtC8HBdKuim3ErGFpMwJLOhUtuJZqQCs4 104
apax/md/ase_calc.py sha256=Z5WKYWVaaKPMLHBPiC5xm1aavqgU_Crh0FLMemHGgJ4 15506
apax/md/constraints.py sha256=Fzvvv0eeLoBB18p2HkDMvGr4owYBOm6DeKFugkS3K7Y 1178
apax/md/dynamics_checks.py sha256=XIPkWMOYy-X33fKKpNJbt4dPhT7y4IvYouM8qT3rx5Q 1496
apax/md/function_transformations.py sha256=TMYnkIbWC3q3qUDuHwnThV2unC9GH6oN13R8mUATSUE 4033
apax/md/io.py sha256=udUwi_wGsHcgqMD9Onm4_FviEvlTjdyGYIGbjSOk8gA 4136
apax/md/md_checkpoint.py sha256=oD0ZLRHaqyHsp2sVWmFKDWSPMhrZ1pz_1TNz4Haz6lg 577
apax/md/schedules.py sha256=2JCXYbNLYABz7BDrZysXmYL7vU8gC1rfuAGTACtPN4s 1592
apax/md/sim_utils.py sha256=ZcB-SafPJJ8cNFPzTKfhhcbNPMltu3cuLFMHTTe9qDM 1155
apax/md/simulate.py sha256=3htBUhQR_cbefc7MyRQDXBBU0z2ccEJGmTAaS8RfcUQ 18124
apax/nn/__init__.py sha256=OHb6Xou2v6u42swTgjRfzej4CIlRg4OmgOIQXUiRjKA 97
apax/nn/__init__.pyi sha256=pYaxSoDiieEWwsAYyBy0zA2UdptTh9hSgPfX3-S63N0 157
apax/nn/builder.py sha256=F3H7Ux6-OI9XK5EnBHJW5b3xeu3r9sz8pLMNz7RB0Jc 8973
apax/nn/models.py sha256=7afg--KUmOB4S9zogRbDRNEXHcEPFwJmE_-weO2-y2U 8864
apax/nodes/__init__.py sha256=QRH9VI35x9uB7I21YLWgHv0OyzFHoRBFEqBjR9MfUck 452
apax/nodes/analysis.py sha256=IvZKiLCwpKWAkMAWhTjVMw44cJ8n7ySqqYp87x90R9Y 1228
apax/nodes/md.py sha256=JJt8NwDAx0MGc4HUokWqsCbqyOQLIN0j8kzllzHsOtE 2313
apax/nodes/model.py sha256=o9QsPhh59BO8Q3VPU2FGV6MmknUDvL74JdytLoVvhiw 9970
apax/nodes/selection.py sha256=i_S2OZPuitkO8-cWSI4mua422VZ8XCddts0GjkhSBDk 8966
apax/nodes/utils.py sha256=O7PREnVrRIKj7aOVr7JnYM8QmRJa5Cd3vjrKw1esQjU 830
apax/optimizer/__init__.py sha256=VHDzmly0IIDdgHJ-vY5pXzF3tRo6rIhgB53C3Lr2-eA 72
apax/optimizer/get_optimizer.py sha256=YEyAhLMMax9FzRBer9kx4AyNArtsFYBvvaSQMBtcews 4406
apax/optimizer/optimizers.py sha256=FrZECvZ1bFmGLtE3J02YrDfJWkQtB4xY2mFYdq7sl7Y 2796
apax/train/__init__.py sha256=po2JhqqjgcFWatxYLKviMNvmKOghBoisykaUbfORh30 163
apax/train/__init__.pyi sha256=UOySKCHWw1-JBecKF4T1HwypRwolHWVBRD3TftZUfvs 178
apax/train/callbacks.py sha256=ALFY--vEMoCrKKWuSros2TQQV5UEQxGZHUGLMWs3NGw 5818
apax/train/checkpoints.py sha256=-X-3plGM1U4RynEd7QOo5zMi4Oz1Q7wPdKPk_OWW1IM 6525
apax/train/eval.py sha256=F9wM846qpWh4hHv6R-Uaj5RB7IUGkzIHa-h0cqSViO4 5859
apax/train/loss.py sha256=ScXnj7LXh11FiYWVGSwSli9x4ABjeX0AdgulH3PHNus 5831
apax/train/metrics.py sha256=wuQDD-DG4jh9g59736dDuNNYzyHY_WoLbf1JMu_Yr1U 2775
apax/train/mlflow.py sha256=sXn-NGdrVlZvwo9iF0kXytdFzwX8X1ZQa_mXilXhEOg 1151
apax/train/parameters.py sha256=CAQRL2bZfl2N2J_XpiTcdAaZ1fI60SjBezkAXfDhMZc 977
apax/train/run.py sha256=gQwOf_mZSlw1L4gIkoNUQVO4PNc43hiuB4WkKW7Zgeg 7562
apax/train/trainer.py sha256=BMq3aWfeHAr1vIDqPHJa2Rk3nVWKlW_5eWpHuNmWOhU 9888
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=3ZxnX74N3VgYkC0Qk86ZiiTJ-g_sVg0LlofMnW-alNM 5027
apax/utils/data.py sha256=rXOdHdVnIdV2RMXVrCFdgtLdyuemz0luvExH4RBAep0 2072
apax/utils/datasets.py sha256=LD3jMRDYk_AoYia3bttGR88Mz3AUKxGWk-8jPVqGoKE 3088
apax/utils/helpers.py sha256=AmsA3n15XMrYgst9DhiD1pmYUer12OlNxd8PuAVcV3s 1625
apax/utils/math.py sha256=u_7jzLV0xokIy6pKVS_tSBoShhCo_9WW4HYhpeE80aU 702
apax/utils/random.py sha256=ShL0hhjgDNhd1hTKWSZHT7HtI6_CztxRXFt9Q8-77UU 170
apax/utils/transform.py sha256=soI688d5E4VwT1mRTxGqi75qkT1roVo7S5V3yItre0k 164
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=XQU2PnoqPkFtRnmFfO6R9dIVzPzRt8FBhCsoexyEJ_E 62040
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-0.10.2.dist-info/METADATA sha256=He2T7ZHUAdRtotnMAkqM4DNmBask4y6_NgGjSJDzGic 5351
apax-0.10.2.dist-info/WHEEL sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8 91
apax-0.10.2.dist-info/entry_points.txt sha256=HmGqoIqJxKFwatM-Z8C5B16QyTFEHDVN27vyaep-E9s 47
apax-0.10.2.dist-info/top_level.txt sha256=SgsiYbsDB_esr6tZ0QNtQwokMY6X6wptGa2fX6IdLGY 5
apax-0.10.2.dist-info/RECORD — —

top_level.txt

apax

entry_points.txt

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