erwincoumans / jax

Related projects ⓘ

Alternatives and complementary repositories for jax

• jcmgray / autoray
  Abstract your array operations.
  ☆136Updated 3 weeks ago
• esa / torchquad
  Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX
  ☆188Updated 4 months ago
• Dipolar-Quantum-Gases / jaxfit
  GPU/TPU accelerated nonlinear least-squares curve fitting using JAX
  ☆51Updated last year
• gradhep / neos
  Upstream optimisation for downstream inference
  ☆69Updated 2 weeks ago
• rfeinman / Torch-ARPACK
  Fast extremal eigensolvers for PyTorch.
  ☆17Updated 3 years ago
• lukasheinrich / pyhep2020-autodiff-tutorial
  ☆31Updated 4 years ago
• crowsonkb / mdmm
  The Modified Differential Multiplier Method (MDMM) for PyTorch
  ☆51Updated 3 years ago
• MilesCranmer / sympy2jax
  Turning SymPy expressions into JAX functions
  ☆42Updated 3 years ago
• buwantaiji / DominantSparseEigenAD
  Reverse-mode AD of dominant sparse eigensolver using Pytorch.
  ☆38Updated 4 years ago
• lsarra / rmi
  Code to estimate Renormalized Mutual Information in simple settings
  ☆12Updated 3 years ago
• carnotresearch / cr-sparse
  Functional models and algorithms for sparse signal processing
  ☆88Updated last year
• danieljtait / jax_xla_adventures
  Dive into Jax, Flax, XLA and C++
  ☆31Updated 4 years ago
• SymJAX / SymJAX
  Documentation:
  ☆120Updated last year
• flatironinstitute / jax-finufft
  JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library
  ☆80Updated this week
• flaport / klujax
  Solve sparse linear systems in JAX using the KLU algorithm
  ☆32Updated this week
• pmelchior / proxmin
  Proximal optimization in pure python
  ☆112Updated 2 years ago
• NIFTy-PPL / JAXbind
  Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation
  ☆61Updated 4 months ago
• Joshuaalbert / jaxns
  Probabilistic Programming and Nested sampling in JAX
  ☆147Updated this week
• ucl-bug / jaxdf
  A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
  ☆120Updated last month
• cornellius-gp / linear_operator
  A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch
  ☆93Updated 2 months ago
• PhilipVinc / numba4jax
  Use numba-compiled kernels from within Jax
  ☆25Updated last month
• ASEM000 / kernex
  Stencil computations in JAX
  ☆66Updated last year
• johannbrehmer / higgs_inference
  Code repository for the paper "Constraining Effective Field Theories with Machine Learning"
  ☆21Updated 5 years ago
• minaskar / zeus
  ⚡️ zeus: Lightning Fast MCMC ⚡️
  ☆227Updated 8 months ago
• AaltoML / kalman-jax
  Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX
  ☆94Updated last year
• imagingofthings / pyFFS
  Fast Fourier Series library.
  ☆19Updated last year
• NeilGirdhar / tjax
  Tools for JAX
  ☆43Updated 3 weeks ago
• CW-Huang / CP-Flow
  Convex potential flows
  ☆78Updated 2 years ago
• andgoldschmidt / derivative
  Optimal numerical differentiation of noisy time series data in python.
  ☆56Updated 2 months ago
• gngdb / pytorch-minimize
  Use scipy.optimize.minimize as a PyTorch Optimizer.
  ☆65Updated 3 months ago