apax

View on PyPIReverse Dependencies (0)

0.9.1 apax-0.9.1-py3-none-any.whl

Wheel Details

Project: apax
Version: 0.9.1
Filename: apax-0.9.1-py3-none-any.whl
Download: [link]
Size: 152525
MD5: ea7c54c5767720583c14f3d37ea9df54
SHA256: 8dad4de289ad3106945c8b6184a0e436a292a4acffab12dcbb5cf6b20c4f87a8
Uploaded: 2024-12-03 16:01:49 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: apax
Version: 0.9.1
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
Classifier: Programming Language :: Python :: 3.12
Classifier: Programming Language :: Python :: 3.13
Requires-Python: >=3.10,<4.0
Requires-Dist: ase (<4.0.0,>=3.22.1)
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: lazy-loader (<0.5,>=0.4)
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 (<3.0.0,>=2.12.0)
Requires-Dist: typer (<0.14,>=0.13)
Requires-Dist: uncertainty-toolbox (<0.2.0,>=0.1.1)
Requires-Dist: vesin (<0.3.0,>=0.2.0)
Requires-Dist: znh5md (<0.5,>=0.4)
Requires-Dist: zntrack (<0.9,>=0.8); extra == "zntrack"
Provides-Extra: mlflow
Provides-Extra: zntrack
Description-Content-Type: text/markdown
[Description omitted; length: 3966 characters]

WHEEL

Wheel-Version: 1.0
Generator: poetry-core 1.9.1
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=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/calibration.py sha256=Xg1w3z3cGj1tI-pOhwKNA-3zYQPKK2fN1uOfyJ3rTDo 2208
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/cli/templates/md_config_minimal.yaml sha256=c-N3Q87jlza7m7-0InQzw5cCmrYak_L-Dt9GCyV35Is 518
apax/cli/templates/train_config_full.yaml sha256=MTKOXKUytvFSRA9HOVO8HR57qJqV7x_5RnUM69KKXvA 2036
apax/cli/templates/train_config_minimal.yaml sha256=7DUkXQC-tan-LtVZUibQwNkg77c5XcfzC0JiQW8Go6o 318
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/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/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/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=hF9A-tw59fDgVK81kKMI9uPIgECNXw6CyKtHjO5xXS8 18090
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=FnXYXYlD21N8ISLstbUM2rvOoKbUmS48QvzgysCBg3s 8956
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=ARpGUh2_g9SwiEl5QTCcNBOOw6xWkrHIPff4kIXZHFg 5822
apax/train/checkpoints.py sha256=MEINK4Ux8twgl7_skrj2nqKmJqtEpAL1kyE9ZfZgFpI 6469
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/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=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/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-0.9.1.dist-info/LICENSE sha256=D1th3_Mod1zsxijBAneLP08ClqxdbQWMZwmm0Kko4Co 1089
apax-0.9.1.dist-info/METADATA sha256=m3omEVJcLb8R-1_ouEewMhlSibL9tE6wFv1Y66zw9v0 5564
apax-0.9.1.dist-info/WHEEL sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs 88
apax-0.9.1.dist-info/entry_points.txt sha256=egIp1X0oJDgpml9cIeC2nmnwi_PAozR198TCswSc3g8 46
apax-0.9.1.dist-info/RECORD

entry_points.txt

apax = apax.cli.apax_app:app