High-performance CUDA kernels for equivariant graph neural networks (MACE, NequIP, Allegro). 10-20x faster than e3nn.
☆44Jan 17, 2026Updated 9 months ago
Alternatives and similar repositories for batmobile
Users that are interested in batmobile are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Building Blocks for Equivariant Neural Networks in e3nn and PyTorch 2.0☆22Nov 16, 2025Updated 11 months ago
- Save pytrees efficiently in hdf5 files.☆15Updated this week
- Pure C implementation of e3nn☆26Mar 17, 2025Updated last year
- generalized machine learning potential eqnorm☆14Jun 30, 2025Updated last year
- Repository for GPU related kernels for learning/testing purposes☆21May 27, 2026Updated 5 months ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- Developer blog for PyTorch☆22Sep 30, 2026Updated last month
- Public releases of the msmJAX fast electrostatics code☆13Mar 26, 2026Updated 7 months ago
- Training Neural Network potentials through customizable routines in JAX.☆74Oct 23, 2026Updated last week
- [ICML'25] The Price of Freedom: Exploring Expressivity and Runtime Tradeoffs in Equivariant Tensor Products☆20Jul 16, 2025Updated last year
- PyTorch operations for use in creating machine learning interatomic potentials☆22Sep 15, 2026Updated last month
- Time-reversal Euclidean neural networks based on e3nn☆22Updated this week
- Resources to learn llvm-ripple programming☆12Updated this week
- An ASE-friendly implementation of the amorphous-to-crystalline (a2c) workflow.☆18Oct 19, 2025Updated last year
- Custom trained GRACE-OFFF models☆12Jun 2, 2026Updated 4 months ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- Robust machine learning interatomic potentials (MLIPs) that achieve accuracy comparable to the ωB97X-D3BJ/def2-TZVPP quantum mechanical m…☆11Jun 9, 2026Updated 4 months ago
- Functional JAX middleware using a compute Node abstraction☆22Sep 6, 2026Updated last month
- PaiNN in jax☆11Jan 14, 2025Updated last year
- This is the source code for paper "Neural Network Potentials for Accelerated Metadynamics of Oxygen Reduction Kinetics at Au-Water Interf…☆24Jun 5, 2024Updated 2 years ago
- OpenEquivariance: a fast, open-source GPU JIT kernel generator for the Clebsch-Gordon Tensor Product.☆172Updated this week
- ☆47Aug 3, 2026Updated 2 months ago
- FLOPS counter for all your GPU benchmarking needs☆13Aug 8, 2024Updated 2 years ago
- ☆28Apr 17, 2026Updated 6 months ago
- Node Equivariant Message Passing Neural Network☆12Jul 4, 2026Updated 3 months ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Grain-boundary generator with Atomman☆13Sep 24, 2022Updated 4 years ago
- ☆17Apr 22, 2025Updated last year
- ☆10Aug 15, 2024Updated 2 years ago
- Implementing PaiNN in Pytorch Geometric☆14Mar 10, 2022Updated 4 years ago
- Real-time collaborative architectural CAD in your browser. Based on Kafka, gRPC, Redis, and OpenCASCADE.☆12Jan 7, 2023Updated 3 years ago
- [NeurIPS 2024] Official implementation of the Efficiently Scaled Attention Interatomic Potential☆60Sep 26, 2025Updated last year
- dftio is to assist machine learning communities to transcript DFT output into a format that is easy to read or used by machine learning m…☆16Dec 18, 2025Updated 10 months ago
- plane wave basis set density functional thoery code☆18Mar 29, 2026Updated 7 months ago
- dataset augmentation for atomistic machine learning☆27Nov 21, 2025Updated 11 months ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- [NeurIPS 2024] source code for "A Recipe for Charge Density Prediction"☆42Dec 17, 2024Updated last year
- Script to generate distorted perovskite structures☆13Feb 10, 2024Updated 2 years ago
- Educational code for Density Functional Theory and Quantum Monte Carlo with Python☆21Mar 29, 2019Updated 7 years ago
- Automatic differentiation for Triton Kernels☆29Aug 12, 2025Updated last year
- Understanding the paper "Principles of Riemannian Geometry in Neural Networks" by Michael Hauser and Asok Ray☆11May 24, 2023Updated 3 years ago
- a mini TPU with floating point arithmetic☆53Dec 22, 2025Updated 10 months ago
- A flexible workflow for on-the-fly learning of interatomic potential models.☆37Oct 8, 2026Updated 3 weeks ago