maxxxzdn / jax-geometric
Implementation of various equivariant models in JAX
☆12Updated 5 months ago
Related projects: ⓘ
- Annotated implementations of equivariant graph neural networks in Jax: EGNN, SEGNN, NequIP.☆17Updated 3 months ago
- Steerable E(3) GNN in jax☆22Updated 11 months ago
- Tools for building equivariant polynomials on reductive Lie groups.☆26Updated last year
- ☆10Updated 9 months ago
- SE(3) Equivariant Augmented Coupling Flows. NeurIPS 2023.☆22Updated 5 months ago
- Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/☆26Updated last year
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆78Updated 2 weeks ago
- C++/CUDA library for SO(3) equivariant operations☆19Updated last week
- An implementation of SchNet in JAX and JAX-MD.☆16Updated 2 years ago
- Higher order equivariant graph neural networks for 3D point clouds☆33Updated last year
- ☆17Updated 5 months ago
- ☆17Updated 10 months ago
- ☆48Updated last year
- Forward mode laplacian implemented in JAX tracer☆27Updated 2 months ago
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆34Updated 2 years ago
- Wigner's 3J, 6J, 9J symbols for python☆18Updated 2 months ago
- Deterministic particle dynamics for simulating Fokker-Planck probability flows☆23Updated last year
- ☆13Updated last year
- PySCF on IPU☆41Updated 6 months ago
- CUDA implementations of MACE models☆10Updated 2 months ago
- ☆27Updated 2 years ago
- Pure C implementation of e3nn☆15Updated last month
- Code repository for "Finding symmetry breaking order parameters with Euclidean Neural Networks"☆12Updated 3 years ago
- Hierarchical variational models for physics.☆18Updated 4 years ago
- Code used by the "Clifford Group Equivariant Neural Networks" paper.☆68Updated 3 months ago
- Equivariant machine learning interatomic potentials in JAX.☆59Updated 11 months ago
- Reference implementation of "Ab-Initio Potential Energy Surfaces by Pairing GNNs with Neural Wave Functions" (ICLR, 2022) and "Sampling-f…☆27Updated 3 months ago
- Molecular mechanics systems and simulation data☆15Updated 3 months ago
- Turn jitted jax functions back into python source code☆20Updated 2 months ago
- A repo to finetune NNIPs by training on observables☆12Updated 5 months ago