heidelberg-hepml / lorentz-gatr
Repository for <Lorentz-Equivariant Geometric Algebra Transformers for High-Energy Physics> (J. Spinner et al 2024)
☆17Updated last week
Alternatives and similar repositories for lorentz-gatr:
Users that are interested in lorentz-gatr are comparing it to the libraries listed below
- Annotated implementations of equivariant graph neural networks in Jax: EGNN, SEGNN, NequIP.☆23Updated last month
- ☆21Updated last year
- Lorentz group equivariant autoencoders based on Lorentz Group Network☆11Updated last year
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆96Updated 2 months ago
- Forward mode laplacian implemented in JAX tracer☆29Updated 2 months ago
- Implementation of Forward Laplacian algorithm in JAX☆57Updated this week
- Steerable E(3) GNN in jax☆24Updated last year
- E(n) Equivariant GNN in jax☆11Updated last year
- Tools for building equivariant polynomials on reductive Lie groups.☆28Updated last year
- Permutation Equivariant Lorentz Invariant/Covariant Aggregator Network☆12Updated 2 months ago
- Machine learning Calabi-Yau metrics with JAX☆17Updated last year
- ESI-DCAFM-TACO-VDSP Summer School on "Machine Learning for Materials Hard and Soft"☆35Updated 2 years ago
- ☆10Updated last year
- Reference implementation of "Ab-Initio Potential Energy Surfaces by Pairing GNNs with Neural Wave Functions" (ICLR, 2022) and "Sampling-f…☆29Updated 7 months ago
- Wigner's 3J, 6J, 9J symbols for python☆19Updated 6 months ago
- Differentiable and accelerated spherical transforms with JAX☆142Updated 3 weeks ago
- [JHEP] An Efficient Lorentz Equivariant Graph Neural Network for Jet Tagging☆23Updated 2 years ago
- Implementation of various equivariant models in JAX☆12Updated 9 months ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆62Updated 6 months ago
- Code for the paper "Algorithmic Differentiation for Automatized Modelling of Machine Learned Force Fields"☆12Updated last year
- Fully and Partially Bayesian Neural Nets☆45Updated last week
- Code used by the "Clifford Group Equivariant Neural Networks" paper.☆79Updated 7 months ago
- Neural network architecture that is fully equivariant with respect to transformations under the Lorentz group, a fundamental symmetry of …☆48Updated 2 years ago
- Official implementation of "Particle Transformer for Jet Tagging".☆88Updated 8 months ago
- Pure C implementation of e3nn☆16Updated last month
- variational free-energy of dense hydrogen☆12Updated last year
- ☆21Updated 5 months ago
- ☆16Updated 2 years ago
- For developing and reproducing ML + HEP projects.☆22Updated last week
- Supporting code for "Autoregressive neural-network wavefunctions for ab initio quantum chemistry".☆39Updated 2 years ago