JAXMg: A multi-GPU linear solver in JAX
☆53Jun 5, 2026Updated this week
Alternatives and similar repositories for jaxmg
Users that are interested in jaxmg are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
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☆34Updated this week
- Code for 'From Tensor Network Quantum States to Tensorial Recurrent Neural Networks'.☆25Feb 11, 2026Updated 3 months ago
- Special functions in Jax☆18Mar 1, 2025Updated last year
- Projected time-dependent Variational Monte Carlo (p-tVMC) method based on infidelity optimization for variational simulation of quantum d…☆18Oct 17, 2024Updated last year
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- NEar-Axis opTimisation☆12Jun 14, 2024Updated last year
- A python code for 2d topology optimization using MMA optimizers in NLOPT☆12Feb 9, 2017Updated 9 years ago
- KiNetic Orbit-averaging SOlver for Stellarators☆16Dec 22, 2023Updated 2 years ago
- Prototype for sparse automatic differentiation in JAX + Julia☆22Jun 17, 2025Updated 11 months ago
- ☆14Oct 11, 2022Updated 3 years ago
- Calculates Boozer coordinates for toroidal MHD equilibria, including stellarators and tokamaks.☆19Jul 29, 2025Updated 10 months ago
- ☆13Jun 10, 2024Updated 2 years ago
- Orthogonal polynomials with JAX☆28May 18, 2026Updated 3 weeks ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆81Feb 12, 2026Updated 3 months ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Python package to reduce the qubit requirements of quantum simulation by embedding into DFT.☆11Feb 27, 2026Updated 3 months ago
- Xtructure is datastructure for using in JAX☆22May 18, 2026Updated 3 weeks ago
- ☆11Sep 16, 2024Updated last year
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆529May 20, 2026Updated 2 weeks ago
- A python implementation of Physics-informed Spline Learning for nonlinear dynamics discovery.☆28Sep 30, 2021Updated 4 years ago
- The next-gen AI-native tensor-network-based quantum software framework☆78Updated this week
- Experiements on how conditional mutual information affects the performance of neural quantum states.☆13Oct 31, 2024Updated last year
- ☆10Oct 30, 2023Updated 2 years ago
- Official TensorFlow implementation of GAN-based priors for quantifying uncertainty.☆12Nov 13, 2021Updated 4 years 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.
- Empowering deep neural quantum states through efficient optimization☆29Jul 8, 2024Updated last year
- an infrastructure wiki. POC☆17Sep 12, 2019Updated 6 years ago
- Code for the paper "Algorithmic Differentiation for Automatized Modelling of Machine Learned Force Fields"☆14Jul 2, 2023Updated 2 years ago
- share NLIGA code which fork form sourceforge☆16Aug 18, 2020Updated 5 years ago
- FLOPS counter for all your GPU benchmarking needs☆13Aug 8, 2024Updated last year
- Python implementation of the Quasisymmetric Stellarator Construction method☆16Jan 25, 2025Updated last year
- ☆58May 27, 2026Updated last week
- Map (deep learning) model weights between different model implementations.☆19Mar 9, 2026Updated 3 months ago
- A MODular development environment and library for OPTimization algorithms☆40Apr 30, 2026Updated last month
- End-to-end encrypted email - Proton Mail • AdSpecial offer: 40% Off Yearly / 80% Off First Month. All Proton services are open source and independently audited for security.
- Symplectic Integration Methods for Particle Loss Estimation☆24Updated this week
- JIT-compiled GPU kernels for quantum chemistry☆34Jan 30, 2026Updated 4 months ago
- Powder X-ray diffraction Rietveld refinement using FullProf.☆12Mar 20, 2022Updated 4 years ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆134Updated this week
- ☆13Updated this week
- Overview of published generative models that produce molecules☆16Jan 16, 2020Updated 6 years ago
- C++ implemented Coordinate Descent Full Configuration Interaction (CDFCI) package for quantum chemistry calculation.☆20May 15, 2026Updated 3 weeks ago