A JAX based package designed for efficient second order operators (e.g., laplacian) computation.
☆75Mar 15, 2024Updated last year
Alternatives and similar repositories for lapjax
Users that are interested in lapjax are comparing it to the libraries listed below
Sorting:
- Efficient and Accurate Neural-Network Ansatz for Quantum Monte Carlo☆72Dec 4, 2024Updated last year
- Forward mode laplacian implemented in JAX tracer☆30Jan 7, 2026Updated last month
- Calculate observables from neural network-based VMC (NN-VMC).☆17Apr 16, 2025Updated 10 months ago
- Implementation of Forward Laplacian algorithm in JAX☆94Nov 25, 2025Updated 3 months ago
- An example showing how to use jax to train resnet50 on multi-node multi-GPU☆20Jul 4, 2022Updated 3 years ago
- A library combining solid quantum Monte Carlo and neural network.☆50Dec 10, 2024Updated last year
- JAX accelerated Quantum Monte Carlo☆92Aug 11, 2025Updated 6 months ago
- A JAX library for Density Functional Theory.☆54Nov 25, 2025Updated 3 months ago
- Flexible, general-purpose VMC framework, built on JAX.☆29Nov 25, 2025Updated 3 months ago
- Python library for real space quantum Monte Carlo☆99Feb 21, 2026Updated last week
- ☆26Jul 30, 2025Updated 7 months ago
- Deep learning quantum Monte Carlo for electrons in real space☆412Jul 14, 2025Updated 7 months ago
- Ab-initio simulation of interacting fermions with equivariant normalizing flow.☆34Oct 23, 2022Updated 3 years ago
- An implementation of the Fermionic Neural Network for ab-initio electronic structure calculations☆817Nov 13, 2025Updated 3 months ago
- Supporting code for "Autoregressive neural-network wavefunctions for ab initio quantum chemistry".☆43Jun 17, 2022Updated 3 years ago
- A Universal Neural Network Solver for The Schrödinger Equation☆11Dec 3, 2024Updated last year
- Impementation of Variational Monte Carlo (VMC) for quantum many-body dynamics using JAX.☆71Sep 26, 2025Updated 5 months ago
- DeepErwin is a python 3.8+ package that implements and optimizes JAX 2.x wave function models for numerical solutions to the multi-electr…☆66Apr 18, 2025Updated 10 months ago
- ☆25Jul 8, 2025Updated 7 months ago
- Code for the paper "Variational Monte Carlo Approach to Partial Differential Equations with Neural Networks" (https://arxiv.org/abs/2206.…☆10Aug 8, 2022Updated 3 years ago
- PyTorch implementation for PaLM: A Hybrid Parser and Language Model.☆10Jan 7, 2020Updated 6 years ago
- Physics-Informed deep LSTM architecture to forecast Lorenz and MFE fluid systems☆13May 18, 2020Updated 5 years ago
- ☆10Jan 16, 2020Updated 6 years ago
- variational free-energy of dense hydrogen☆14Sep 25, 2023Updated 2 years ago
- Ab Initio Energies☆10Nov 22, 2025Updated 3 months ago
- Exchange correlation functionals translated from libxc to jax☆52Mar 24, 2025Updated 11 months ago
- The source code the for the ICLR'24 paper "Stabilizing Backpropagation Through Time to Learn Complex Physics"☆11May 17, 2024Updated last year
- The codes calculate energy spectrum, Chern number, particle partition entanglement and quantum metric of the band of the twisted bilayer …☆14Oct 27, 2024Updated last year
- Fast low-rank updates (LRU) of matrix determinants and pfaffians in JAX☆19Feb 6, 2026Updated 3 weeks ago
- Code for 'From Tensor Network Quantum States to Tensorial Recurrent Neural Networks'.☆24Feb 11, 2026Updated 3 weeks ago
- ☆12Nov 30, 2022Updated 3 years ago
- ☆17Sep 16, 2025Updated 5 months ago
- This repository contains the source code for the VFK algorithm introduced in ArXiv:2204.03454☆16Jun 10, 2025Updated 8 months ago
- ☆15Nov 20, 2023Updated 2 years ago
- Physcial Informed Extreme Learning Machine(PIELM) method to solve PDEs, such as Possion problem☆17Dec 6, 2024Updated last year
- Repository for material accompanying the ptvmc systematic study paper☆16Nov 17, 2025Updated 3 months ago
- Pauli operators manipulation toolbox.☆17Updated this week
- Experiements on how conditional mutual information affects the performance of neural quantum states.☆13Oct 31, 2024Updated last year
- Implementation of Koopmans-compliant functionals in Quantum ESPRESSO☆17Jan 23, 2026Updated last month