jackhcollins / EMD_VAE
Variational Autoencoder for Particle Physics using Earth-Movers Distance
☆13Updated 2 years ago
Related projects ⓘ
Alternatives and complementary repositories for EMD_VAE
- Things that make me feel productive☆15Updated 2 years ago
- Tutorial Code for MLHEP pyprob☆18Updated last year
- For developing and reproducing ML + HEP projects.☆22Updated this week
- ☆20Updated 3 months ago
- Provides differentiable versions of common HEP operations and objectives.☆24Updated last year
- Lorentz group equivariant autoencoders based on Lorentz Group Network☆10Updated last year
- A collection of graph neural networks implementations in JAX☆31Updated 11 months ago
- ☆16Updated 2 years ago
- Gradient Based Nested Sampling☆14Updated 9 months ago
- The message passing GAN https://arxiv.org/abs/2106.11535 and generative adversarial particle transformer https://arxiv.org/abs/2211.10295…☆13Updated last month
- ☆31Updated 4 years ago
- Neural network architecture that is fully equivariant with respect to transformations under the Lorentz group, a fundamental symmetry of …☆47Updated 2 years ago
- ☆21Updated last year
- Python package for the EnergyFlow suite of tools.☆41Updated last week
- Transformer-guided diffusion for galaxy clustering. Code repository associated with https://arxiv.org/abs/2311.17141☆21Updated 10 months ago
- Annotated implementations of equivariant graph neural networks in Jax: EGNN, SEGNN, NequIP.☆20Updated 4 months ago
- Probabilistic modeling of tabular data with normalizing flows.☆55Updated 4 months ago
- Code to estimate Renormalized Mutual Information in simple settings☆12Updated 3 years ago
- Implementation of the jet classification network in ParticleNet: Jet Tagging via Particle Clouds☆47Updated 4 years ago
- ☆19Updated 2 years ago
- ☆13Updated last year
- A package to describe amortized (conditional) normalizing-flow PDFs defined jointly on tensor products of manifolds with coverage control…☆42Updated 9 months ago
- Steerable E(3) GNN in jax☆22Updated last year
- ☆48Updated 2 years ago
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Updated 3 years ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆61Updated 4 months ago
- Sliced Iterative Generator (SIG) & Gaussianizing Iterative Slicing (GIS)☆35Updated last year
- differentiable (binned) likelihoods with JAX☆18Updated last week
- Jet tagging in the Lund plane with graph networks☆8Updated 2 years ago
- Differentiable and accelerated spherical transforms with JAX☆134Updated 2 weeks ago