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
Sorting:
- Repository for <Lorentz-Equivariant Geometric Algebra Transformers for High-Energy Physics> (J. Spinner et al 2024)☆25Updated this week
- A python library for studying Calabi-Yau metrics☆26Updated 9 months ago
- Stencil computations in JAX☆70Updated last year
- Multiple dispatch over abstract array types in JAX.☆121Updated last week
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆92Updated this week
- Interpolation and function approximation with JAX☆177Updated this week
- madjax☆14Updated 5 months ago
- Orthogonal polynomials with JAX☆22Updated last week
- Numerical quadrature with JAX☆58Updated last week
- ☆31Updated 4 years ago
- Matrix-free linear algebra in JAX.☆118Updated 4 months ago
- Machine learning–based inference toolkit for particle physics☆86Updated 6 months ago
- Provides differentiable versions of common HEP operations and objectives.☆24Updated last year
- Simulation-based inference in JAX☆31Updated last month
- Jax-based quantum state vector simulator.☆26Updated 2 weeks ago
- Normalizing-flow enhanced sampling package for probabilistic inference in Jax☆233Updated 2 weeks ago
- Turning SymPy expressions into JAX functions☆45Updated 4 years ago
- A Python package for efficient optimisation of real-space renormalization group transformations using Tensorflow.☆29Updated 2 months ago
- differentiable (binned) likelihoods with JAX☆22Updated this week
- Fast, lightweight and parallelised simulation-based inference in JAX.☆18Updated this week
- Things that make me feel productive☆15Updated 2 years ago
- Automatic parametric modeling with symbolic regression☆46Updated last week
- Automatic Functional Differentiation in JAX☆72Updated last month
- Forward mode laplacian implemented in JAX tracer☆29Updated 2 months ago
- ☆21Updated last year
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆107Updated 2 months ago
- A differentiable cosmology library in JAX☆196Updated 4 months ago
- Code for the paper "Algorithmic Differentiation for Automatized Modelling of Machine Learned Force Fields"☆12Updated last year
- A generic interface for linear algebra backends☆73Updated 2 months ago
- Neural network architecture that is fully equivariant with respect to transformations under the Lorentz group, a fundamental symmetry of …☆50Updated 3 years ago