ml4physics / cyjax
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
- A python library for studying Calabi-Yau metrics☆26Updated 8 months ago
- Numerical quadrature with JAX☆56Updated last week
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆90Updated last week
- ☆21Updated last year
- madjax☆14Updated 5 months ago
- Interpolation and function approximation with JAX☆172Updated last week
- Repository for <Lorentz-Equivariant Geometric Algebra Transformers for High-Energy Physics> (J. Spinner et al 2024)☆23Updated 2 weeks ago
- A differentiable cosmology library in JAX☆195Updated 4 months ago
- Normalizing-flow enhanced sampling package for probabilistic inference in Jax☆230Updated this week
- ☆31Updated 4 years ago
- Neural network architecture that is fully equivariant with respect to transformations under the Lorentz group, a fundamental symmetry of …☆50Updated 3 years ago
- Automatic Functional Differentiation in JAX☆71Updated 3 weeks ago
- Fast, lightweight and parallelised simulation-based inference in JAX.☆18Updated this week
- Differentiable and accelerated spherical transforms with JAX☆147Updated this week
- Multiple dispatch over abstract array types in JAX.☆116Updated last week
- Probabilistic Programming and Nested sampling in JAX☆174Updated last week
- A system for scientific simulation-based inference at scale.☆163Updated last year
- Community-sourced list of papers and resources on neural simulation-based inference.☆114Updated 2 months ago
- Simulation-based inference benchmark☆96Updated 2 months ago
- Simulation-based inference in JAX☆31Updated last week
- Things that make me feel productive☆14Updated 2 years ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆67Updated 2 weeks ago
- Stencil computations in JAX☆70Updated last year
- ☆155Updated last month
- MCHMC: sampler from an arbitrary differentiable distribution☆72Updated last month
- Normalizing Flows using JAX☆83Updated last year
- Matrix-free linear algebra in JAX.☆116Updated 3 months ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆103Updated last year
- Machine learning–based inference toolkit for particle physics☆86Updated 5 months ago
- Provides differentiable versions of common HEP operations and objectives.☆24Updated last year