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:
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆97Updated last week
- A python library for studying Calabi-Yau metrics☆26Updated 11 months ago
- Matrix-free linear algebra in JAX.☆125Updated last month
- Numerical quadrature with JAX☆65Updated last week
- Orthogonal polynomials with JAX☆22Updated last week
- Minimal JAX implementation of k-nearest neighbors using a k-d tree.☆41Updated last week
- Stencil computations in JAX☆71Updated last year
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆72Updated 2 months ago
- Repository for <Lorentz-Equivariant Geometric Algebra Transformers for High-Energy Physics> (J. Spinner et al 2024)☆26Updated 2 weeks ago
- madjax☆14Updated 7 months ago
- Simulation-based inference benchmark☆97Updated 5 months ago
- Multiple dispatch over abstract array types in JAX.☆125Updated last week
- Simulation-based inference in JAX☆32Updated 2 months ago
- Automatic Functional Differentiation in JAX☆72Updated 3 months ago
- Turning SymPy expressions into JAX functions☆45Updated 4 years ago
- Community-sourced list of papers and resources on neural simulation-based inference.☆118Updated last month
- Probabilistic Programming and Nested sampling in JAX☆184Updated 2 weeks ago
- Fast, lightweight and parallelised simulation-based inference in JAX.☆18Updated last week
- ☆31Updated 4 years ago
- Python package for computing indirect detection constraints on sub-GeV dark matter.☆12Updated 4 months ago
- Notebooks for "A high bias low-variance introduction to Machine Learning for physicists."☆69Updated 5 years ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆111Updated 3 months ago
- A differentiable cosmology library in JAX☆203Updated 6 months ago
- Exact OU processes with JAX☆48Updated 3 months ago
- Exponential families for JAX☆71Updated this week
- Wigner's 3J, 6J, 9J symbols for python☆21Updated 11 months ago
- JAX bindings for the NVIDIA cuDecomp library☆37Updated 2 weeks ago
- S2FFT: Differentiable and accelerated spherical transforms☆195Updated this week
- A system for scientific simulation-based inference at scale.☆164Updated last year
- ☆22Updated last year