jackhcollins / EMD_VAELinks
Variational Autoencoder for Particle Physics using Earth-Movers Distance
☆13Updated 3 years ago
Alternatives and similar repositories for EMD_VAE
Users that are interested in EMD_VAE are comparing it to the libraries listed below
Sorting:
- ☆23Updated last year
- Tutorial Code for MLHEP pyprob☆18Updated 2 years ago
- For developing and reproducing ML + HEP projects.☆24Updated 3 weeks ago
- ☆16Updated 3 years ago
- Things that make me feel productive☆15Updated 3 years ago
- ☆22Updated 2 years ago
- Neural network architecture that is fully equivariant with respect to transformations under the Lorentz group, a fundamental symmetry of …☆51Updated 3 years ago
- A collection of graph neural networks implementations in JAX☆35Updated last year
- E(n) Equivariant GNN in jax☆14Updated 2 years ago
- Differentiable scientific computing library☆155Updated last year
- ☆33Updated 5 years ago
- Python package for the EnergyFlow suite of tools.☆42Updated 2 months ago
- ☆14Updated 2 years ago
- SciPy-based kd-tree with periodic boundary conditions☆37Updated 9 years ago
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Updated 3 years ago
- Deterministic particle dynamics for simulating Fokker-Planck probability flows☆26Updated 2 years ago
- Steerable E(3) GNN in jax☆24Updated 2 years ago
- Code to estimate Renormalized Mutual Information in simple settings☆13Updated 4 years ago
- Code repository for "Finding symmetry breaking order parameters with Euclidean Neural Networks"☆16Updated 4 years ago
- Implementation of the jet classification network in ParticleNet: Jet Tagging via Particle Clouds☆49Updated 5 years ago
- Provides differentiable versions of common HEP operations and objectives.☆25Updated 2 years ago
- ☆53Updated 3 years ago
- ☆19Updated 3 years ago
- ☆12Updated 2 years ago
- A package to describe amortized (conditional) normalizing-flow PDFs defined jointly on tensor products of manifolds with coverage control…☆46Updated 7 months ago
- Graph Convolutional Networks in JAX☆33Updated 4 years ago
- Library for normalizing flows and neural flows.☆25Updated 3 years ago
- Code for the paper "Variational Monte Carlo Approach to Partial Differential Equations with Neural Networks" (https://arxiv.org/abs/2206.…☆10Updated 3 years ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆74Updated last month
- Flow Annealed Importance Sampling Bootstrap (FAB) with JAX.☆12Updated last year