microsoft / folx
Implementation of Forward Laplacian algorithm in JAX
☆61Updated last month
Alternatives and similar repositories for folx:
Users that are interested in folx are comparing it to the libraries listed below
- Forward mode laplacian implemented in JAX tracer☆29Updated last month
- Exchange correlation functionals translated from libxc to jax☆45Updated last week
- Multiple dispatch over abstract array types in JAX.☆115Updated this week
- Flow Annealed Importance Sampling Bootstrap (FAB) with JAX.☆10Updated 9 months ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆90Updated this week
- ☆49Updated 2 years ago
- Reference implementation of "Ab-Initio Potential Energy Surfaces by Pairing GNNs with Neural Wave Functions" (ICLR, 2022) and "Sampling-f…☆30Updated 9 months ago
- Stencil computations in JAX☆70Updated last year
- Automatic Functional Differentiation in JAX☆68Updated last week
- Implementation of various equivariant models in JAX☆12Updated 11 months ago
- Ab-initio simulation of interacting fermions with equivariant normalizing flow.☆34Updated 2 years ago
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆103Updated 4 months ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆45Updated 2 months ago
- Turning SymPy expressions into JAX functions☆44Updated 4 years ago
- Flexible, general-purpose VMC framework, built on JAX.☆26Updated last week
- ☆18Updated 3 months ago
- Matrix-free linear algebra in JAX.☆116Updated 2 months ago
- PySCF on IPU☆42Updated last year
- Stainless neural networks in JAX☆33Updated last week
- ☆11Updated last year
- Steerable E(3) GNN in jax☆24Updated last year
- A dead simple Python package for creating custom JAX pytree objects☆63Updated 4 months ago
- JAX accelerated Quantum Monte Carlo☆73Updated 3 weeks ago
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆30Updated last month
- JAX bindings for the NVIDIA cuDecomp library☆32Updated last month
- jax library for E3 Equivariant Neural Networks☆202Updated 2 months ago
- Things that make me feel productive☆14Updated 2 years 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 3 months ago
- variational free-energy of dense hydrogen☆12Updated last year
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆66Updated last week