flatironinstitute / jaxmgView external linksLinks
JAXMg: A multi-GPU linear solver in JAX
☆35Jan 27, 2026Updated 2 weeks ago
Alternatives and similar repositories for jaxmg
Users that are interested in jaxmg are comparing it to the libraries listed below
Sorting:
- Benchmark scripts for comparing tutorials in PyTorch and JAX☆14Aug 25, 2022Updated 3 years ago
- Flexible and scalable neural quantum states built on JAX☆27Feb 7, 2026Updated last week
- A package to simplify working with symmetry-adapted quantum many-body bases. Provides a good foundation for writing custom exact diagonal…☆34Jan 11, 2026Updated last month
- NEar-Axis opTimisation☆12Jun 14, 2024Updated last year
- ☆10Oct 30, 2023Updated 2 years ago
- A python code for 2d topology optimization using MMA optimizers in NLOPT☆12Feb 9, 2017Updated 9 years ago
- Fast low-rank updates (LRU) of matrix determinants and pfaffians in JAX☆18Feb 6, 2026Updated last week
- KiNetic Orbit-averaging SOlver for Stellarators☆14Dec 22, 2023Updated 2 years ago
- Code for the paper "Algorithmic Differentiation for Automatized Modelling of Machine Learned Force Fields"☆12Jul 2, 2023Updated 2 years ago
- ☆14Oct 11, 2022Updated 3 years ago
- ☆11Sep 16, 2024Updated last year
- Powder X-ray diffraction Rietveld refinement using FullProf.☆12Mar 20, 2022Updated 3 years ago
- Prototype for sparse automatic differentiation in JAX + Julia☆20Jun 17, 2025Updated 7 months ago
- SParse lINear solvers in JAX☆26Feb 2, 2026Updated 2 weeks ago
- A Python package for analyzing the Hofstadter model.☆19Jun 10, 2024Updated last year
- Python implementation of the Quasisymmetric Stellarator Construction method☆16Jan 25, 2025Updated last year
- Matlab codes for hotspot analysis to avoid overheating in Topology optimization for Additive Manufacturing☆11Dec 12, 2025Updated 2 months ago
- share NLIGA code which fork form sourceforge☆15Aug 18, 2020Updated 5 years ago
- e-Stellarator Simulation and Optimization Suite☆14Jan 30, 2026Updated 2 weeks ago
- ☆13Feb 3, 2026Updated last week
- Official TensorFlow implementation of GAN-based priors for quantifying uncertainty.☆13Nov 13, 2021Updated 4 years ago
- Progress toward fusion energy breakeven and gain as measured against the Lawson criterion, figures and data☆18Dec 16, 2025Updated 2 months ago
- A differentiable, flexible stellarator coil proxy☆15Jan 22, 2026Updated 3 weeks ago
- Experiements on how conditional mutual information affects the performance of neural quantum states.☆13Oct 31, 2024Updated last year
- A MODular development environment and library for OPTimization algorithms☆31Jan 14, 2026Updated last month
- FLOPS counter for all your GPU benchmarking needs☆13Aug 8, 2024Updated last year
- Repository to create easy and straightforward slides in Manim together with Manim-slides☆19Nov 26, 2025Updated 2 months ago
- ☆14Jul 23, 2022Updated 3 years ago
- Tensorflow implementation of message passing neural networks for molecules and materials☆12Mar 9, 2020Updated 5 years ago
- an infrastructure wiki. POC☆17Sep 12, 2019Updated 6 years ago
- Overview of published generative models that produce molecules☆15Jan 16, 2020Updated 6 years ago
- Projected time-dependent Variational Monte Carlo (p-tVMC) method based on infidelity optimization for variational simulation of quantum d…☆17Oct 17, 2024Updated last year
- PyTorch-based auto-differentiable orbital-free density functional theory package☆13Mar 19, 2024Updated last year
- Python package for handling n-port data (Touchstone, CITI)☆21Sep 25, 2023Updated 2 years ago
- Special functions in Jax☆16Mar 1, 2025Updated 11 months ago
- Julia-based framework for battery modelling☆16Feb 18, 2025Updated 11 months ago
- ☆15Mar 20, 2023Updated 2 years ago
- JIT-compiled GPU kernels for quantum chemistry☆31Jan 30, 2026Updated 2 weeks ago
- A minimal block sparse symmetric and fermionic tensor python library☆28Updated this week