lab-cosmo / sparse_accumulation
☆12Updated last year
Alternatives and similar repositories for sparse_accumulation:
Users that are interested in sparse_accumulation are comparing it to the libraries listed below
- ☆23Updated last year
- ☆21Updated last year
- Equivariant machine learning interatomic potentials in JAX.☆72Updated 2 weeks ago
- A library for building equivariant neural networks and a zoo of implementations & examples.☆32Updated 2 years ago
- MESS: Modern Electronic Structure Simulations☆20Updated 7 months ago
- Learning neural network potentials from experimental data via Differentiable Trajectory Reweighting☆31Updated last year
- Higher order equivariant graph neural networks for 3D point clouds☆37Updated last year
- OpenEquivariance: a fast, open-source GPU JIT kernel generator for the Clebsch-Gordon Tensor Product.☆57Updated this week
- ☆18Updated 2 years ago
- [TMLR 2025] Stability-Aware Training of Machine Learning Force Fields with Differentiable Boltzmann Estimators☆13Updated 2 months ago
- This repo contains the datasets used for the paper The Design Space of E(3)-Equivariant Atom-Centered Interatomic Potentials.☆18Updated 2 years ago
- [NeurIPS 2024] source code for "A Recipe for Charge Density Prediction"☆33Updated 4 months ago
- Multi-language library for the calculation of spherical harmonics in Cartesian coordinates☆84Updated this week
- Force-field-enhanced Neural Networks optimized library☆42Updated this week
- Particle-mesh based calculations of long-range interactions in PyTorch☆47Updated last week
- C++/CUDA library for SO(3) equivariant operations☆23Updated 2 weeks ago
- Particle-mesh based calculations of long-range interactions in JAX☆17Updated 2 months ago
- Compute neighbor lists for atomistic systems☆53Updated 2 weeks ago
- MESS: Modern Electronic Structure Simulations☆28Updated last month
- ☆21Updated last year
- Deprecated - see `pair_nequip_allegro`☆43Updated last week
- CUDA implementations of MACE models☆14Updated 3 months ago
- Training Neural Network potentials through customizable routines in JAX.☆25Updated 2 months ago
- Training and evaluating machine learning models for atomistic systems.☆31Updated this week
- A collection of simulation recipes for the atomic-scale modeling of materials and molecules☆22Updated this week
- Lab materials for MIT 3.320 and Harvard AP 275 courses on atomistic modeling☆15Updated 4 years ago
- A Newtonian message passing network for deep learning of interatomic potentials and forces☆43Updated last week
- Jax / Haiku implementation of DimeNet++.☆18Updated 3 years ago
- Predict the electronic structure and atomic properties (potential energy, forces, and stress tensor) of polymers containing N and/or O.☆19Updated 8 months ago
- A lightweight DeepPotentialMD with JAX backend, and more than that! Built for both performance and flexibility in pure Python.☆29Updated last week