☆51Aug 19, 2026Updated last month
Alternatives and similar repositories for tojax
Users that are interested in tojax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- A high-performance toolkit for atomistic simulations in JAX.☆234Updated this week
- A Python package for performing Widom insertion simulations to calculate gas adsorption properties in porous materials☆19Sep 22, 2025Updated 11 months ago
- [ICML'26] Phonon fine-tuning (PFT) and [NeurIPS'25 AI4Mat] Nequix: Training a foundation model for materials on a budget☆76Apr 5, 2026Updated 5 months ago
- Official repository of Zatom-1, a general-purpose model for 3D molecules and materials☆53Jul 25, 2026Updated last month
- ALCHEMI Toolkit-Ops is a collection of optimized batch kernels to accelerate computational chemistry and material science workflows.☆231Updated this week
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- An ASE-friendly implementation of the amorphous-to-crystalline (a2c) workflow.☆18Oct 19, 2025Updated 11 months ago
- Learning Hamiltonian Flow Maps via Mean Flow Consistency for Large-Timestep Molecular Dynamics☆40Jul 4, 2026Updated 2 months ago
- A molecular simulation package integrating MLFFs in MOFs for DAC☆48Oct 17, 2025Updated 11 months ago
- ALCHEMI Toolkit is a developer toolkit for accelerating training and inference for AI in chemistry and material science.☆167Updated this week
- JAX implementation of the NequIP neural network interatomic potential☆17Feb 24, 2026Updated 6 months ago
- [NeurIPS 2025 Spotlight] E2Former: An Efficient and Equivariant Transformer with Linear-Scaling Tensor Products☆31Sep 2, 2026Updated 2 weeks ago
- Compute neighbor lists for atomistic systems☆85Sep 10, 2026Updated last week
- ☆23Aug 18, 2026Updated last month
- Official implementation of "Neural Pfaffians: Solving Many Many-Electron Schrödinger Equations"☆15Jul 29, 2026Updated last month
- Simple, predictable pricing with DigitalOcean hosting • AdAlways know what you'll pay with monthly caps and flat pricing. Enterprise-grade infrastructure trusted by 600k+ customers.
- HIP: Hessians with Interatomic Potentials☆42Sep 8, 2026Updated last week
- E(n) Equivariant GNN in jax☆14Aug 31, 2023Updated 3 years ago
- 🌟 [NeurIPS '25 Spotlight] Fair and transparent benchmark of machine learning interatomic potentials (MLIPs), beyond basic error metrics …☆109Jul 2, 2026Updated 2 months ago
- Train, fine-tune, and manipulate machine learning models for atomistic systems☆79Updated this week
- A universal ML model to predict molecular dynamics trajectories with long time steps☆57Sep 9, 2026Updated last week
- [NeurIPS 2024] Official implementation of the Efficiently Scaled Attention Interatomic Potential☆60Sep 26, 2025Updated 11 months ago
- ☆18Jul 29, 2024Updated 2 years ago
- 🌟 [NeurIPS '25 Spotlight] Official implement of QHFlow for DFT Hamiltonian prediction☆32Dec 6, 2025Updated 9 months ago
- graph2mat: Graph to matrix conversion☆25Sep 7, 2026Updated last week
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Python package containing a variety of distance functions for crystals, as well as evaluation metrics for crystal generation.☆32Jun 22, 2026Updated 2 months ago
- Zero Shot Molecular Generation via Similarity Kernels☆31Aug 30, 2026Updated 2 weeks ago
- Lightweight diffusion transformer for crystalline materials discovery☆27Updated this week
- Library for Crystal Symmetry in Rust☆81Updated this week
- Generative materials benchmarking metrics, inspired by guacamol and CDVAE.☆41Jun 1, 2024Updated 2 years ago
- SevenNet - a graph neural network interatomic potential package supporting efficient multi-GPU parallel molecular dynamics simulations.☆273Jul 22, 2026Updated last month
- ML potentials via transfer learning☆30Sep 8, 2026Updated last week
- Infrastructure to implement and train NNPs☆28Sep 3, 2026Updated 2 weeks ago
- train and use graph-based ML models of potential energy surfaces☆131Updated this week
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Quick tools for materials chemistry☆19May 29, 2024Updated 2 years ago
- CrystalGRW: Generative Modeling of Crystal Structures with Targeted Properties via Geodesic Random Walks☆15May 22, 2026Updated 3 months ago
- Library for efficient training and application of Machine Learning Interatomic Potentials (MLIP)☆135Aug 4, 2026Updated last month
- Torch-native, batchable, atomistic simulations.☆494Updated this week
- ☆73Jul 9, 2026Updated 2 months ago
- An evaluation framework for machine learning models simulating high-throughput materials discovery.☆252Updated this week
- Framework for modified sampling from biomolecular generative models☆27Updated this week