☆50Aug 19, 2026Updated last week
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.☆229Updated 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 4 months ago
- Official repository of Zatom-1, a general-purpose model for 3D molecules and materials☆51Jul 25, 2026Updated last month
- ALCHEMI Toolkit-Ops is a collection of optimized batch kernels to accelerate computational chemistry and material science workflows.☆226Updated this week
- 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.
- An ASE-friendly implementation of the amorphous-to-crystalline (a2c) workflow.☆18Oct 19, 2025Updated 10 months ago
- Learning Hamiltonian Flow Maps via Mean Flow Consistency for Large-Timestep Molecular Dynamics☆40Jul 4, 2026Updated last month
- A molecular simulation package integrating MLFFs in MOFs for DAC☆48Oct 17, 2025Updated 10 months ago
- ALCHEMI Toolkit is a developer toolkit for accelerating training and inference for AI in chemistry and material science.☆150Updated 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☆30Feb 16, 2026Updated 6 months ago
- Compute neighbor lists for atomistic systems☆85Aug 7, 2026Updated 3 weeks ago
- ☆22Aug 18, 2026Updated last week
- Official implementation of "Neural Pfaffians: Solving Many Many-Electron Schrödinger Equations"☆15Jul 29, 2026Updated last month
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- HIP: Hessians with Interatomic Potentials☆41Aug 19, 2026Updated last week
- E(n) Equivariant GNN in jax☆14Aug 31, 2023Updated 2 years ago
- 🌟 [NeurIPS '25 Spotlight] Fair and transparent benchmark of machine learning interatomic potentials (MLIPs), beyond basic error metrics …☆105Jul 2, 2026Updated last month
- Train, fine-tune, and manipulate machine learning models for atomistic systems☆78Updated this week
- A universal ML model to predict molecular dynamics trajectories with long time steps☆57Jul 5, 2026Updated last month
- [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 8 months ago
- graph2mat: Graph to matrix conversion☆24Aug 4, 2026Updated 3 weeks ago
- 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.
- 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 27, 2025Updated last year
- Lightweight diffusion transformer for crystalline materials discovery☆26Jul 26, 2026Updated last month
- Library for Crystal Symmetry in Rust☆77Updated 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.☆270Jul 22, 2026Updated last month
- ML potentials via transfer learning☆30Jul 28, 2026Updated last month
- Infrastructure to implement and train NNPs☆28Jul 30, 2026Updated last month
- train and use graph-based ML models of potential energy surfaces☆129Jul 13, 2026Updated last month
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- 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)☆131Aug 4, 2026Updated 3 weeks ago
- Torch-native, batchable, atomistic simulations.☆487Aug 21, 2026Updated last week
- ☆71Jul 9, 2026Updated last month
- An evaluation framework for machine learning models simulating high-throughput materials discovery.☆248Aug 23, 2026Updated last week
- Framework for modified sampling from biomolecular generative models☆25Updated this week