heidelberg-hepml / lorentz-gatr
Repository for <Lorentz-Equivariant Geometric Algebra Transformers for High-Energy Physics> (J. Spinner et al 2024)
☆12Updated this week
Related projects ⓘ
Alternatives and complementary repositories for lorentz-gatr
- Annotated implementations of equivariant graph neural networks in Jax: EGNN, SEGNN, NequIP.☆21Updated this week
- cuEquivariance is a math library that is a collective of low-level primitives and tensor ops to accelerate widely-used models, like DiffD…☆53Updated this week
- ☆21Updated last year
- Tools for building equivariant polynomials on reductive Lie groups.☆26Updated last year
- Code used by the "Clifford Group Equivariant Neural Networks" paper.☆76Updated 5 months ago
- Equivariant machine learning interatomic potentials in JAX.☆62Updated last year
- jax library for E3 Equivariant Neural Networks☆184Updated last month
- Pure C implementation of e3nn☆15Updated this week
- C++/CUDA library for SO(3) equivariant operations☆19Updated last month
- ☆13Updated last year
- Implementation of various equivariant models in JAX☆12Updated 7 months ago
- Transformer-guided diffusion for galaxy clustering. Code repository associated with https://arxiv.org/abs/2311.17141☆21Updated 10 months ago
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆92Updated last week
- ☆21Updated 7 months ago
- ESI-DCAFM-TACO-VDSP Summer School on "Machine Learning for Materials Hard and Soft"☆34Updated 2 years ago
- Permutation Equivariant Lorentz Invariant/Covariant Aggregator Network☆12Updated 3 weeks ago
- Forward mode laplacian implemented in JAX tracer☆29Updated 2 weeks ago
- Code for the paper "Algorithmic Differentiation for Automatized Modelling of Machine Learned Force Fields"☆12Updated last year
- Steerable E(3) GNN in jax☆22Updated last year
- Neural network architecture that is fully equivariant with respect to transformations under the Lorentz group, a fundamental symmetry of …☆48Updated 2 years ago
- Higher order equivariant graph neural networks for 3D point clouds☆33Updated last year
- CUDA implementations of MACE models☆10Updated 4 months ago
- Lorentz group equivariant autoencoders based on Lorentz Group Network☆10Updated last year
- SE(3) Equivariant Augmented Coupling Flows. NeurIPS 2023.☆23Updated 7 months ago
- Machine learning Calabi-Yau metrics with JAX☆15Updated last year
- A JAX library for Density Functional Theory.☆42Updated 6 months ago
- For developing and reproducing ML + HEP projects.☆22Updated 2 weeks ago
- Differentiable and accelerated spherical transforms with JAX☆136Updated this week
- ☆18Updated last year
- MESS: Modern Electronic Structure Simulations☆20Updated last month