iaifi / summer-school-2023Links
☆22Updated 2 years ago
Alternatives and similar repositories for summer-school-2023
Users that are interested in summer-school-2023 are comparing it to the libraries listed below
Sorting:
- ☆14Updated 2 years ago
- Repository for <Lorentz-Equivariant Geometric Algebra Transformers for High-Energy Physics> (J. Spinner et al 2024)☆29Updated 4 months ago
- Transformer-guided diffusion for galaxy clustering. Code repository associated with https://arxiv.org/abs/2311.17141☆25Updated last year
- E(n) Equivariant GNN in jax☆14Updated 2 years ago
- Graph Neural Networks to predict the cosmological parameters or the galaxy power spectrum from galaxy catalogs.☆18Updated 2 years ago
- ☆23Updated last year
- ☆16Updated 3 years ago
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆39Updated 7 months ago
- ☆34Updated 2 years ago
- ☆39Updated last year
- Normalizing-flow enhanced sampling package for probabilistic inference in Jax☆250Updated last month
- Neural network architecture that is fully equivariant with respect to transformations under the Lorentz group, a fundamental symmetry of …☆51Updated 3 years ago
- jax library for E3 Equivariant Neural Networks☆216Updated last month
- Things that make me feel productive☆15Updated 3 years ago
- Permutation Equivariant Lorentz Invariant/Covariant Aggregator Network☆13Updated 11 months ago
- ☆19Updated 3 years ago
- A Jax wrapper for cudaKDTree☆11Updated 2 weeks ago
- Simulation-based inference in JAX☆35Updated 5 months ago
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆109Updated 6 months ago
- Code used by the "Clifford Group Equivariant Neural Networks" paper.☆84Updated last year
- Gradient Based Nested Sampling☆20Updated 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…☆59Updated 5 months ago
- Python calculator for the Wigner D matrices, Clebsch-Gordan coefficients, hydrogen atoms, and irreducible representations of the Lorentz …☆10Updated 2 years ago
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.☆20Updated last year
- A collection of graph neural networks implementations in JAX☆35Updated last year
- Probabilistic modeling of tabular data with normalizing flows.☆58Updated 2 weeks ago
- Community-sourced list of papers and resources on neural simulation-based inference.☆128Updated 4 months ago
- A package to describe amortized (conditional) normalizing-flow PDFs defined jointly on tensor products of manifolds with coverage control…☆46Updated 6 months ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆101Updated this week
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆74Updated last week