rkansal47 / MPGAN
The message passing GAN https://arxiv.org/abs/2106.11535 and generative adversarial particle transformer https://arxiv.org/abs/2211.10295 architectures for generating particle clouds
☆13Updated 2 months ago
Related projects: ⓘ
- For developing and reproducing ML + HEP projects.☆22Updated 2 weeks ago
- Lorentz group equivariant autoencoders based on Lorentz Group Network☆10Updated last year
- Variational Autoencoder for Particle Physics using Earth-Movers Distance☆13Updated 2 years ago
- Implementation of the jet classification network in ParticleNet: Jet Tagging via Particle Clouds☆44Updated 4 years ago
- LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite☆55Updated 2 months ago
- A package to describe amortized (conditional) normalizing-flow PDFs defined jointly on tensor products of manifolds with coverage control…☆42Updated 7 months ago
- Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"☆46Updated last year
- ☆21Updated last year
- Things that make me feel productive☆15Updated last year
- ☆31Updated 3 years ago
- Probabilistic programming framework for signal inference algorithms that operate regardless of the underlying grids and their resolutions☆16Updated this week
- ☆12Updated 9 months ago
- JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework☆48Updated 3 weeks ago
- PyTorch examples for NERSC systems☆28Updated last year
- [JHEP] An Efficient Lorentz Equivariant Graph Neural Network for Jet Tagging☆23Updated 2 years ago
- ☆18Updated 10 months ago
- Convolutional Differential Operators for Physics-based Deep Learning Study☆21Updated last month
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆61Updated 2 months ago
- Running PDEs on graph☆27Updated last year
- ☆33Updated last year
- JAX bindings for the NVIDIA cuDecomp library☆24Updated 2 weeks ago
- Simulator Expansion for Likelihood-Free Inference (SELFI): a python implementation☆10Updated last year
- Code for Mesh Transformer describes in the EAGLE dataset☆28Updated 5 months ago
- Python package for the EnergyFlow suite of tools.☆40Updated last month
- Transformer-guided diffusion for galaxy clustering. Code repository associated with https://arxiv.org/abs/2311.17141☆20Updated 8 months ago
- Fork of the E2CNN library that adds support for equivariant partial differential operators☆17Updated 2 years ago
- Config files for my GitHub profile.☆22Updated 8 months ago
- Differentiable and accelerated spherical transforms with JAX☆123Updated this week
- Machine learning Calabi-Yau metrics with JAX☆15Updated last year
- ☆21Updated last year