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
- ☆21Updated 8 months ago
- Lorentz group equivariant autoencoders based on Lorentz Group Network☆11Updated last year
- Tutorial Code for MLHEP pyprob☆18Updated last year
- A package to describe amortized (conditional) normalizing-flow PDFs defined jointly on tensor products of manifolds with coverage control…☆43Updated 3 weeks ago
- ☆16Updated 2 years ago
- For developing and reproducing ML + HEP projects.☆22Updated this week
- ☆21Updated last year
- Python package for the EnergyFlow suite of tools.☆42Updated 2 months ago
- ☆31Updated 4 years ago
- Code to estimate Renormalized Mutual Information in simple settings☆13Updated 4 years ago
- ☆19Updated 3 years ago
- Provides differentiable versions of common HEP operations and objectives.☆24Updated last year
- ☆24Updated last year
- Gradient Based Nested Sampling☆20Updated last year
- Transformer-guided diffusion for galaxy clustering. Code repository associated with https://arxiv.org/abs/2311.17141☆23Updated last year
- ☆13Updated last year
- Flow Annealed Importance Sampling Bootstrap (FAB) with JAX.☆11Updated 10 months ago
- A matplotlib style sheet to help me spend less time editing figures for my manuscripts for Phys. Rev. D.☆19Updated 3 months ago
- Probabilistic modeling of tabular data with normalizing flows.☆56Updated 3 weeks ago
- Jet tagging in the Lund plane with graph networks☆8Updated 3 years ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆67Updated this week
- Probabilistic programming framework for signal inference algorithms that operate regardless of the underlying grids and their resolutions☆26Updated 3 weeks ago
- A Jax wrapper for cudaKDTree☆10Updated last year
- GATSBI: Generative Adversarial Training for Simulation-Based Inference☆18Updated last year
- Implementation of the jet classification network in ParticleNet: Jet Tagging via Particle Clouds☆47Updated 4 years ago
- E(n) Equivariant GNN in jax☆11Updated last year
- Fast, lightweight and parallelised simulation-based inference in JAX.☆18Updated last week
- JAX bindings for the NVIDIA cuDecomp library☆32Updated 2 weeks ago
- Neural network architecture that is fully equivariant with respect to transformations under the Lorentz group, a fundamental symmetry of …☆50Updated 3 years ago