ml4physics / cyjaxLinks
Machine learning Calabi-Yau metrics with JAX
☆17Updated last year
Alternatives and similar repositories for cyjax
Users that are interested in cyjax are comparing it to the libraries listed below
Sorting:
- Repository for <Lorentz-Equivariant Geometric Algebra Transformers for High-Energy Physics> (J. Spinner et al 2024)☆25Updated this week
- Numerical quadrature with JAX☆61Updated last month
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆95Updated this week
- A python library for studying Calabi-Yau metrics☆26Updated 10 months ago
- Stencil computations in JAX☆71Updated last year
- Orthogonal polynomials with JAX☆22Updated this week
- Multiple dispatch over abstract array types in JAX.☆124Updated this week
- Simulation-based inference in JAX☆31Updated last month
- Provides differentiable versions of common HEP operations and objectives.☆24Updated last year
- Minimal JAX implementation of k-nearest neighbors using a k-d tree.☆37Updated this week
- Fast, lightweight and parallelised simulation-based inference in JAX.☆18Updated 2 weeks ago
- differentiable (binned) likelihoods with JAX☆22Updated this week
- Community-sourced list of papers and resources on neural simulation-based inference.☆116Updated 2 weeks ago
- Normalizing-flow enhanced sampling package for probabilistic inference in Jax☆239Updated last week
- Interpolation and function approximation with JAX☆190Updated 3 weeks ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆72Updated last month
- Simulation-based inference benchmark☆97Updated 4 months ago
- Matrix-free linear algebra in JAX.☆119Updated last week
- ☆21Updated last year
- Probabilistic Programming and Nested sampling in JAX☆182Updated this week
- A system for scientific simulation-based inference at scale.☆164Updated last year
- MCHMC: sampler from an arbitrary differentiable distribution☆73Updated last month
- madjax☆14Updated 6 months ago
- A differentiable cosmology library in JAX☆199Updated 5 months ago
- Things that make me feel productive☆15Updated 2 years ago
- Automatic Functional Differentiation in JAX☆72Updated 2 months ago
- Probabilistic programming framework for signal inference algorithms that operate regardless of the underlying grids and their resolutions☆30Updated last week
- Python package for computing indirect detection constraints on sub-GeV dark matter.☆12Updated 4 months ago
- ☆161Updated 3 weeks ago
- Exact OU processes with JAX☆48Updated 2 months ago