Dependents of e3nn-jax

17 dependents

Package	Description	Downloads/month
jax-md	Differentiable, Hardware Accelerated, Molecular Dynamics	95K
nequix	[ICML'26] Phonon fine-tuning (PFT) and [NeurIPS'25 AI4Mat] Nequix: Training a fo...	85K
skalax	JAX implementation of the Skala neural exchange-correlation functional	1K
mlip	Library for efficient training and application of Machine Learning Interatomic P...	894
harmonyx	Fully-Differentiable Tensor Spherical Harmonics in JAX	536
tensor-harmonix	Fully-differentiable tensor spherical harmonics in JAX	504
tensorial	Library for machine learning on physical tensors	419
lagrangebench	LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite	222
e3md	Library for machine learning on physical tensors	148
e3response	Library for machine learning on physical tensors	139
learnax	Training Pipeline Library	100
deepbind	Protein+Ligand Accelerated Simulation	98
deepjump	Accelerated Protein Dynamics Simulation	95
ophiuchus	Protein Autoencoding	92
latentjump	Protein Accelerated Latent Simulation	77
tensorclouds	Training Pipeline Library	60
nequip-eqx	JAX implementation of the NequIP neural network interatomic potential	56