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:
- Repository for <Lorentz-Equivariant Geometric Algebra Transformers for High-Energy Physics> (J. Spinner et al 2024)☆29Updated 5 months ago
- ☆33Updated 5 years ago
- Matrix-free linear algebra in JAX.☆151Updated last week
- Abstract your array operations.☆160Updated this week
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆106Updated last week
- Implementation of Forward Laplacian algorithm in JAX☆92Updated 2 weeks ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆117Updated last month
- Minimal JAX implementation of k-nearest neighbors using a k-d tree.☆50Updated 4 months ago
- Jax-based quantum state vector simulator.☆27Updated 5 months ago
- A python library for studying Calabi-Yau metrics☆27Updated 4 months ago
- Stencil computations in JAX☆71Updated 3 weeks ago
- Exact OU processes with JAX☆57Updated 8 months ago
- Orthogonal polynomials with JAX☆26Updated this week
- Interpolation and function approximation with JAX☆231Updated last week
- Things that make me feel productive☆15Updated 3 years ago
- Turn SymPy expressions into trainable JAX expressions.☆355Updated 7 months ago
- 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
- Multiple dispatch over abstract array types in JAX.☆136Updated 2 weeks ago
- madjax☆14Updated last year
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆211Updated 4 months ago
- Machine learning–based inference toolkit for particle physics☆86Updated last year
- Neural network architecture that is fully equivariant with respect to transformations under the Lorentz group, a fundamental symmetry of …☆51Updated 3 years ago
- Numerical quadrature with JAX☆73Updated this week
- Python package for computing indirect detection constraints on sub-GeV dark matter.☆13Updated 10 months ago
- S2FFT: Differentiable and accelerated spherical transforms☆208Updated this week
- Forward mode laplacian implemented in JAX tracer☆29Updated 2 months ago
- Provides differentiable versions of common HEP operations and objectives.☆25Updated 2 years ago
- differentiable (binned) likelihoods with JAX☆24Updated last week
- Error propagation and statistical analysis for Monte Carlo simulations in lattice QCD and statistical mechanics using autograd.☆50Updated 2 weeks ago
- Mathematical operations for JAX pytrees☆205Updated last year