jackhcollins / EMD_VAE
Variational Autoencoder for Particle Physics using Earth-Movers Distance
☆13Updated 2 years ago
Alternatives and similar repositories for EMD_VAE:
Users that are interested in EMD_VAE are comparing it to the libraries listed below
- Things that make me feel productive☆14Updated 2 years ago
- ☆31Updated 4 years ago
- Tutorial Code for MLHEP pyprob☆18Updated last year
- ☆22Updated 7 months ago
- ☆16Updated 2 years ago
- ☆21Updated last year
- Lorentz group equivariant autoencoders based on Lorentz Group Network☆11Updated last year
- ☆13Updated last year
- For developing and reproducing ML + HEP projects.☆22Updated 3 weeks ago
- Fast, lightweight and parallelised simulation-based inference in JAX.☆17Updated 2 weeks ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆66Updated this week
- Repository for <Lorentz-Equivariant Geometric Algebra Transformers for High-Energy Physics> (J. Spinner et al 2024)☆22Updated 3 weeks ago
- Provides differentiable versions of common HEP operations and objectives.☆24Updated last year
- Transformer-guided diffusion for galaxy clustering. Code repository associated with https://arxiv.org/abs/2311.17141☆23Updated last year
- Using Graph Neural Networks to regress baryonic properties directly from full dark matter merger trees.☆24Updated last year
- Graph Neural Networks to predict the cosmological parameters or the galaxy power spectrum from galaxy catalogs.☆17Updated 2 years ago
- Jet tagging in the Lund plane with graph networks☆8Updated 3 years ago
- A package to describe amortized (conditional) normalizing-flow PDFs defined jointly on tensor products of manifolds with coverage control…☆43Updated this week
- Differentiable and accelerated spherical transforms with JAX☆146Updated this week
- Python package for the EnergyFlow suite of tools.☆42Updated 2 months ago
- Library for normalizing flows and neural flows.☆24Updated 2 years ago
- Implementation of the CWoLa Hunting strategy for dijet resonances☆18Updated 4 years ago
- Machine learning–based inference toolkit for particle physics☆86Updated 4 months ago
- Implementation of the jet classification network in ParticleNet: Jet Tagging via Particle Clouds☆46Updated 4 years ago
- Probabilistic modeling of tabular data with normalizing flows.☆55Updated last month
- A JAX package for weak gravitational lensing☆31Updated 11 months ago
- Using neural networks to extract sufficient statistics from data by maximising the Fisher information☆31Updated last year
- The message passing GAN https://arxiv.org/abs/2106.11535 and generative adversarial particle transformer https://arxiv.org/abs/2211.10295…☆13Updated 2 months ago
- A Jax wrapper for cudaKDTree☆10Updated last year
- A collection of graph neural networks implementations in JAX☆32Updated last year