An implementation of SchNet in JAX and JAX-MD.
☆17Apr 5, 2022Updated 4 years ago
Alternatives and similar repositories for schnax
Users that are interested in schnax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- tools for graph-based machine-learning potentials in jax☆28Apr 9, 2024Updated 2 years ago
- Jax / Haiku implementation of DimeNet++.☆18Mar 31, 2022Updated 4 years ago
- Reference implementation of "Ewald-based Long-Range Message Passing for Molecular Graphs" (ICML 2023)☆52May 4, 2026Updated 6 months ago
- Metadynamics code on the G-space.☆17Jan 14, 2026Updated 9 months ago
- Code for the paper "Algorithmic differentiation for plane-wave DFT"☆11Mar 4, 2026Updated 8 months ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Build neural networks for machine learning force fields with JAX☆138Jun 2, 2025Updated last year
- ☆24Mar 18, 2024Updated 2 years ago
- SOAP (Smooth Overlap of Atomic Positions) implementation with numpy☆11May 11, 2018Updated 8 years ago
- Collection of Quantum Mechanical Dataset Relevant for Biomolecules☆12Oct 16, 2026Updated 3 weeks ago
- Particle-mesh based calculations of long-range interactions in JAX☆32Updated this week
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆128Oct 20, 2026Updated 2 weeks ago
- ☆14May 10, 2024Updated 2 years ago
- A flexible and performant framework for training machine learning potentials.☆39Updated this week
- Learning neural network potentials from experimental data via Differentiable Trajectory Reweighting☆38Feb 29, 2024Updated 2 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- ☆25Nov 1, 2023Updated 3 years ago
- [TMLR 2023] Training and simulating MD with ML force fields☆116Oct 30, 2024Updated 2 years ago
- Code for the paper "Algorithmic Differentiation for Automatized Modelling of Machine Learned Force Fields"☆14Jul 2, 2023Updated 3 years ago
- jax library for E3 Equivariant Neural Networks☆235Apr 1, 2026Updated 7 months ago
- Public releases of the NeuralIL differentiable neural-network force field☆14Sep 10, 2024Updated 2 years ago
- I beg to Differ☆14Sep 2, 2026Updated 2 months ago
- Codebase for Cormorant Neural Networks☆61May 11, 2022Updated 4 years ago
- ☆10Mar 31, 2023Updated 3 years ago
- GradDFT is a JAX-based library enabling the differentiable design and experimentation of exchange-correlation functionals using machine l…☆116Feb 13, 2024Updated 2 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Syntax aware code search for Julia☆15Jun 22, 2026Updated 4 months ago
- Machine Learned Interatomic Potential Tools☆25Updated this week
- Universal PET models in JAX☆12Oct 29, 2026Updated last week
- MACE_Osaka24 models☆30Apr 15, 2026Updated 6 months ago
- ☆10Sep 19, 2023Updated 3 years ago
- Rapid building, fitting, and application of machine learning force fields☆10Oct 19, 2026Updated 2 weeks ago
- UF3: a python library for generating ultra-fast interatomic potentials☆74Oct 23, 2026Updated 2 weeks ago
- SO3krates and Universal Pairwise Force Field for Molecular Simulation☆257Updated this week
- ☆14Feb 11, 2025Updated last year
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Multi-language library for the calculation of spherical harmonics in Cartesian coordinates☆105Sep 14, 2026Updated last month
- ☆14Jun 11, 2026Updated 4 months ago
- EquiTriton is a project that seeks to implement high-performance kernels for commonly used building blocks in equivariant neural networks…☆77Oct 15, 2026Updated 3 weeks ago
- Moment Invariants Local Atomic Descriptor☆35Aug 20, 2024Updated 2 years ago
- Adaptive sampling algorithms for molecular transitions☆30Jun 28, 2026Updated 4 months ago
- ☆53Oct 27, 2022Updated 4 years ago
- Official Implememtation of FEAT: Free energy Estimators with Adaptive Transport☆16Sep 12, 2026Updated last month