YWolfeee / lapjax
A JAX based package designed for efficient second order operators (e.g., laplacian) computation.
☆71Updated 11 months ago
Alternatives and similar repositories for lapjax:
Users that are interested in lapjax are comparing it to the libraries listed below
- Efficient and Accurate Neural-Network Ansatz for Quantum Monte Carlo☆56Updated 2 months ago
- JAX accelerated Quantum Monte Carlo☆71Updated 2 months ago
- A JAX library for Density Functional Theory.☆47Updated 9 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…☆54Updated 2 months ago
- Reference implementation of "Ab-Initio Potential Energy Surfaces by Pairing GNNs with Neural Wave Functions" (ICLR, 2022) and "Sampling-f…☆29Updated 8 months ago
- Machine learning for physicists☆119Updated last year
- Forward mode laplacian implemented in JAX tracer☆29Updated 3 months ago
- ☆14Updated last year
- Exchange correlation functionals translated from libxc to jax☆43Updated last year
- ☆241Updated 5 months ago
- Code for the paper Universal Physics Transformers☆89Updated last week
- [ICLR 2024] Neural Spectral Methods: Self-supervised learning in the spectral domain.☆36Updated last year
- Mathematical Introduction to Electronic Structure Theory☆45Updated 6 years ago
- PySCF with auto-differentiation☆76Updated this week
- Code for "DPOT: Auto-Regressive Denoising Operator Transformer for Large-Scale PDE Pre-Training"☆49Updated 8 months ago
- Space Group Informed Transformer for Crystalline Materials Generation☆61Updated 6 months ago
- ESI-DCAFM-TACO-VDSP Summer School on "Machine Learning for Materials Hard and Soft"☆35Updated 2 years ago
- Spectral Neural Operator☆75Updated last year
- ☆69Updated 5 months ago
- ICON for in-context operator learning☆45Updated 2 months ago
- [ICLR 2024] Scaling physics-informed hard constraints with mixture-of-experts.☆28Updated 8 months ago
- ☆67Updated 5 months ago
- [ICLR 2024 Spotlight] Official Implementation of "Enabling Efficient Equivariant Operations in the Fourier Basis via Gaunt Tensor Product…☆57Updated 3 months ago
- Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"☆49Updated last year
- A library combining solid quantum Monte Carlo and neural network.☆37Updated 2 months ago
- Python library for real space quantum Monte Carlo☆87Updated last month
- Automatic Functional Differentiation in JAX☆63Updated last month
- Supporting code for "Autoregressive neural-network wavefunctions for ab initio quantum chemistry".☆41Updated 2 years ago
- Official implementation of Scalable Transformer for PDE surrogate modelling☆36Updated 10 months ago
- jax library for E3 Equivariant Neural Networks☆199Updated 3 weeks ago