ml4physics / cyjaxLinks
Machine learning Calabi-Yau metrics with JAX
☆17Updated 2 years ago
Alternatives and similar repositories for cyjax
Users that are interested in cyjax are comparing it to the libraries listed below
Sorting:
- Numerical quadrature with JAX☆73Updated last week
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆106Updated 2 weeks ago
- Minimal JAX implementation of k-nearest neighbors using a k-d tree.☆50Updated 4 months ago
- Things that make me feel productive☆15Updated 3 years ago
- Matrix-free linear algebra in JAX.☆149Updated 2 weeks ago
- Repository for <Lorentz-Equivariant Geometric Algebra Transformers for High-Energy Physics> (J. Spinner et al 2024)☆29Updated 5 months ago
- madjax☆14Updated last year
- A differentiable cosmology library in JAX☆216Updated 4 months ago
- Fast, lightweight and parallelised simulation-based inference in JAX.☆22Updated 3 weeks ago
- ☆33Updated 5 years ago
- S2FFT: Differentiable and accelerated spherical transforms☆209Updated this week
- Orthogonal polynomials with JAX☆26Updated last week
- Implementation of Forward Laplacian algorithm in JAX☆89Updated last month
- Interpolation and function approximation with JAX☆230Updated last week
- This is project page for the paper "RG-Flow: a hierarchical and explainable flow model based on renormalization group and sparse prior". …☆89Updated 3 years ago
- A system for scientific simulation-based inference at scale.☆164Updated last year
- Normalizing-flow enhanced sampling package for probabilistic inference in Jax☆252Updated 3 weeks ago
- Machine learning–based inference toolkit for particle physics☆86Updated last year
- ☆22Updated 2 years ago
- Jax-based quantum state vector simulator.☆26Updated 4 months ago
- Python package for the EnergyFlow suite of tools.☆42Updated 3 months ago
- Stencil computations in JAX☆71Updated this week
- A python library for studying Calabi-Yau metrics☆27Updated 4 months ago
- Simulation-based inference in JAX☆40Updated 7 months ago
- Abstract your array operations.☆160Updated 2 weeks ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆74Updated last month
- Probabilistic programming framework for signal inference algorithms that operate regardless of the underlying grids and their resolutions☆39Updated this week
- Error propagation and statistical analysis for Monte Carlo simulations in lattice QCD and statistical mechanics using autograd.☆50Updated 2 weeks ago
- Exact OU processes with JAX☆56Updated 8 months ago
- differentiable (binned) likelihoods with JAX☆24Updated last week