microsoft / folx
Implementation of Forward Laplacian algorithm in JAX
☆55Updated last week
Related projects ⓘ
Alternatives and complementary repositories for folx
- Forward mode laplacian implemented in JAX tracer☆29Updated last week
- Multiple dispatch over abstract array types in JAX.☆102Updated this week
- ☆48Updated 2 years ago
- Exchange correlation functionals translated from libxc to jax☆43Updated 11 months ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆80Updated this week
- PySCF on IPU☆42Updated 8 months ago
- Reference implementation of "Ab-Initio Potential Energy Surfaces by Pairing GNNs with Neural Wave Functions" (ICLR, 2022) and "Sampling-f…☆28Updated 4 months ago
- Automatic Functional Differentiation in JAX☆57Updated 8 months ago
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆89Updated 3 weeks ago
- Stencil computations in JAX☆66Updated last year
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆41Updated last month
- ☆67Updated this week
- Implementation of various equivariant models in JAX☆12Updated 7 months ago
- variational free-energy of dense hydrogen☆11Updated last year
- Differentiable and accelerated spherical transforms with JAX☆135Updated this week
- jax library for E3 Equivariant Neural Networks☆183Updated last month
- Tools for building equivariant polynomials on reductive Lie groups.☆26Updated last year
- DeepErwin is a python 3.8+ package that implements and optimizes JAX 2.x wave function models for numerical solutions to the multi-electr…☆51Updated 5 months ago
- Turn jitted jax functions back into python source code☆20Updated 4 months ago
- A dead simple Python package for creating custom JAX pytree objects☆61Updated 4 months ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆61Updated 4 months ago
- ☆103Updated 2 weeks ago
- Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/☆29Updated last year
- ☆10Updated 10 months ago
- JAX Arrays for human consumption☆88Updated last year
- Equivariant machine learning interatomic potentials in JAX.☆61Updated last year
- Riemannian Optimization Using JAX☆45Updated last year
- Stainless neural networks in JAX☆30Updated 3 weeks ago
- Arbitrary-order derivatives of popular electronic structure methods, such as Hartree-Fock and coupled cluster theory.☆62Updated 2 months ago
- A collection of graph neural networks implementations in JAX☆31Updated 11 months ago