iaifi / summer-school-2023
☆21Updated last year
Alternatives and similar repositories for summer-school-2023:
Users that are interested in summer-school-2023 are comparing it to the libraries listed below
- ☆13Updated last year
- Transformer-guided diffusion for galaxy clustering. Code repository associated with https://arxiv.org/abs/2311.17141☆23Updated last year
- Repository for <Lorentz-Equivariant Geometric Algebra Transformers for High-Energy Physics> (J. Spinner et al 2024)☆22Updated 2 weeks ago
- ☆22Updated 7 months ago
- ☆16Updated 2 years ago
- Graph Neural Networks to predict the cosmological parameters or the galaxy power spectrum from galaxy catalogs.☆17Updated 2 years ago
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.☆15Updated last year
- [JHEP] An Efficient Lorentz Equivariant Graph Neural Network for Jet Tagging☆23Updated 2 years ago
- Gradient Based Nested Sampling☆20Updated last year
- Permutation Equivariant Lorentz Invariant/Covariant Aggregator Network☆13Updated 4 months ago
- Differentiable Cosmological Forward Model☆78Updated 2 weeks ago
- ☆18Updated 2 weeks ago
- JAX-powered Cosmological Particle-Mesh N-body Solver☆34Updated 3 weeks ago
- JAX-based linear Einstein-Boltzmann solver for cosmology☆14Updated last week
- JAX bindings for the NVIDIA cuDecomp library☆32Updated 2 weeks ago
- ☆34Updated last year
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆66Updated 8 months ago
- E(n) Equivariant GNN in jax☆11Updated last year
- madjax☆14Updated 4 months ago
- Gravitational-wave data analysis tools in Jax☆66Updated last week
- Using Graph Neural Networks to regress baryonic properties directly from full dark matter merger trees.☆24Updated last year
- Numerical quadrature with JAX☆53Updated this week
- A package to describe amortized (conditional) normalizing-flow PDFs defined jointly on tensor products of manifolds with coverage control…☆43Updated 3 weeks ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆89Updated 2 weeks ago
- Simulation-based inference in JAX☆31Updated last month
- A Jax wrapper for cudaKDTree☆10Updated last year
- Demonstration of MomentNetworks for high-dimensional probability density estimation (LFI)☆15Updated 2 years ago
- Tutorials for the KITP Galevo23 program☆26Updated last year
- Implemenation of PQMass from Lemos et al. 2024☆13Updated 4 months ago