13 dependents
Package Description Downloads/month
Differentiable, Hardware Accelerated, Molecular Dynamics 95K
[ICML'26] Phonon fine-tuning (PFT) and [NeurIPS'25 AI4Mat] Nequix: Training a fo... 85K
Python library for RAD Embeddings, provably correct latent DFA representations. 3K
Library for efficient training and application of Machine Learning Interatomic P... 894
Library for machine learning on physical tensors 419
Deep Learning for Bayesian Inference (DL4BI) 321
JAX-SPH: Smoothed Particle Hydrodynamics in JAX 236
LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite 222
A python implementation of the Fast Stokesian Dynamics methods (A. Fiore, J. Swa... 219
Geometric morphometrics in non-Euclidean shape spaces 183
Library for machine learning on physical tensors 148
Library for machine learning on physical tensors 139
JAX implementation of the NequIP neural network interatomic potential 56