Alternatives and similar repositories for cyjax

Users that are interested in cyjax are comparing it to the libraries listed below

• flatironinstitute / jax-finufft
  JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library
  ☆97Updated last week
• pythoncymetric / cymetric
  A python library for studying Calabi-Yau metrics
  ☆26Updated 11 months ago
• pnkraemer / matfree
  Matrix-free linear algebra in JAX.
  ☆125Updated last month
• f0uriest / quadax
  Numerical quadrature with JAX
  ☆65Updated last week
• f0uriest / orthax
  Orthogonal polynomials with JAX
  ☆22Updated last week
• dodgebc / jaxkd
  Minimal JAX implementation of k-nearest neighbors using a k-d tree.
  ☆41Updated last week
• ASEM000 / kernex
  Stencil computations in JAX
  ☆71Updated last year
• NIFTy-PPL / JAXbind
  Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation
  ☆72Updated 2 months ago
• heidelberg-hepml / lorentz-gatr
  Repository for <Lorentz-Equivariant Geometric Algebra Transformers for High-Energy Physics> (J. Spinner et al 2024)
  ☆26Updated 2 weeks ago
• madjax-hep / madjax
  madjax
  ☆14Updated 7 months ago
• sbi-benchmark / sbibm
  Simulation-based inference benchmark
  ☆97Updated 5 months ago
• patrick-kidger / quax
  Multiple dispatch over abstract array types in JAX.
  ☆125Updated last week
• dirmeier / sbijax
  Simulation-based inference in JAX
  ☆32Updated 2 months ago
• sail-sg / autofd
  Automatic Functional Differentiation in JAX
  ☆72Updated 3 months ago
• MilesCranmer / sympy2jax
  Turning SymPy expressions into JAX functions
  ☆45Updated 4 years ago
• smsharma / awesome-neural-sbi
  Community-sourced list of papers and resources on neural simulation-based inference.
  ☆118Updated last month
• Joshuaalbert / jaxns
  Probabilistic Programming and Nested sampling in JAX
  ☆184Updated 2 weeks ago
• homerjed / sbiax
  Fast, lightweight and parallelised simulation-based inference in JAX.
  ☆18Updated last week
• lukasheinrich / pyhep2020-autodiff-tutorial
  ☆31Updated 4 years ago
• LoganAMorrison / Hazma
  Python package for computing indirect detection constraints on sub-GeV dark matter.
  ☆12Updated 4 months ago
• Emergent-Behaviors-in-Biology / mlreview_notebooks
  Notebooks for "A high bias low-variance introduction to Machine Learning for physicists."
  ☆69Updated 5 years ago
• cornellius-gp / linear_operator
  A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch
  ☆111Updated 3 months ago
• DifferentiableUniverseInitiative / jax_cosmo
  A differentiable cosmology library in JAX
  ☆203Updated 6 months ago
• normal-computing / thermox
  Exact OU processes with JAX
  ☆48Updated 3 months ago
• NeilGirdhar / efax
  Exponential families for JAX
  ☆71Updated this week
• fujiisoup / py3nj
  Wigner's 3J, 6J, 9J symbols for python
  ☆21Updated 11 months ago
• DifferentiableUniverseInitiative / jaxDecomp
  JAX bindings for the NVIDIA cuDecomp library
  ☆37Updated 2 weeks ago
• astro-informatics / s2fft
  S2FFT: Differentiable and accelerated spherical transforms
  ☆195Updated this week
• undark-lab / swyft
  A system for scientific simulation-based inference at scale.
  ☆164Updated last year
• iaifi / summer-school-2023
  ☆22Updated last year