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:
- A Lorentz-Equivariant Transformer for All of the LHC☆29Updated 7 months ago
- Orthogonal polynomials with JAX☆26Updated this week
- ☆34Updated 5 years ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆109Updated 2 weeks ago
- Minimal JAX implementation of k-nearest neighbors using a k-d tree.☆52Updated 6 months ago
- Numerical quadrature with JAX☆75Updated this week
- Matrix-free linear algebra in JAX.☆160Updated last month
- Implementation of Forward Laplacian algorithm in JAX☆93Updated last month
- madjax☆14Updated last year
- Stencil computations in JAX☆71Updated 2 months ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆123Updated last month
- Fast, lightweight and parallelised simulation-based inference in JAX.☆22Updated 2 months ago
- S2FFT: Differentiable and accelerated spherical transforms☆212Updated this week
- Interpolation and function approximation with JAX☆240Updated this week
- A differentiable cosmology library in JAX☆221Updated 6 months ago
- Things that make me feel productive☆15Updated 3 years ago
- Exact OU processes with JAX☆59Updated last week
- Jax-based quantum state vector simulator.☆25Updated 6 months ago
- Python package for the EnergyFlow suite of tools.☆42Updated 5 months ago
- Python package for computing indirect detection constraints on sub-GeV dark matter.☆13Updated last month
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆211Updated 5 months ago
- differentiable (binned) likelihoods with JAX☆25Updated this week
- Wigner's 3J, 6J, 9J symbols for python☆22Updated 6 months ago
- Simulation-based inference benchmark☆108Updated 11 months ago
- A python library for studying Calabi-Yau metrics☆27Updated 6 months ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆75Updated 3 months ago
- This is project page for the paper "RG-Flow: a hierarchical and explainable flow model based on renormalization group and sparse prior". …☆88Updated 3 years ago
- Probabilistic Programming and Nested sampling in JAX☆218Updated last month
- Abstract your array operations.☆163Updated last month
- This is a Pytorch implementation of [normalizing flows on tori and spheres, ICML 2020]☆19Updated 3 years ago