Alternatives and similar repositories for alan-cluster

Users that are interested in alan-cluster are comparing it to the libraries listed below

• francois-rozet / dawgz
  Unleash the true power of scheduling
  ☆33Updated 7 months ago
• glouppe / talk-template
  Template for talks in remark+KaTeX.
  ☆25Updated 5 years ago
• francois-rozet / inox
  Stainless neural networks in JAX
  ☆34Updated 2 months ago
• DifferentiableUniverseInitiative / jax_cosmo
  A differentiable cosmology library in JAX
  ☆215Updated 4 months ago
• kazewong / flowMC
  Normalizing-flow enhanced sampling package for probabilistic inference in Jax
  ☆251Updated 2 months ago
• f0uriest / interpax
  Interpolation and function approximation with JAX
  ☆226Updated last week
• dodgebc / jaxkd
  Minimal JAX implementation of k-nearest neighbors using a k-d tree.
  ☆49Updated 3 months ago
• undark-lab / swyft
  A system for scientific simulation-based inference at scale.
  ☆164Updated last year
• mpi4jax / mpi4jax
  Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python
  ☆498Updated 2 weeks ago
• bmad-sim / Bmad-X
  Experimental Bmad Code transcribed in Python with Numba and Pytorch
  ☆13Updated last year
• danielward27 / flowjax
  ☆196Updated last week
• cornellius-gp / linear_operator
  A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch
  ☆116Updated last month
• alonfnt / bayex
  Minimal Implementation of Bayesian Optimization in JAX
  ☆99Updated 6 months ago
• JaxGaussianProcesses / GPJax
  Gaussian processes in JAX and Flax.
  ☆548Updated this week
• google-deepmind / distrax
  ☆602Updated this week
• google-deepmind / kfac-jax
  Second Order Optimization and Curvature Estimation with K-FAC in JAX.
  ☆294Updated 2 weeks ago
• Joshuaalbert / jaxns
  Probabilistic Programming and Nested sampling in JAX
  ☆210Updated 2 months ago
• jeremiecoullon / jax-tqdm
  Add a tqdm progress bar to your JAX scans and loops.
  ☆118Updated 5 months ago
• patrick-kidger / sympy2jax
  Turn SymPy expressions into trainable JAX expressions.
  ☆352Updated 6 months ago
• dfm / tinygp
  The tiniest of Gaussian Process libraries
  ☆323Updated 3 weeks ago
• astro-informatics / s2fft
  S2FFT: Differentiable and accelerated spherical transforms
  ☆208Updated this week
• esa / torchquad
  Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX
  ☆209Updated 3 months ago
• probabilists / lampe
  Likelihood-free AMortized Posterior Estimation with PyTorch
  ☆131Updated last year
• johannbrehmer / pointers
  Things that make me feel productive
  ☆15Updated 3 years ago
• jax-ml / oryx
  Oryx is a library for probabilistic programming and deep learning built on top of Jax.
  ☆281Updated 3 weeks ago
• patrick-kidger / optimistix
  Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
  ☆487Updated this week
• flatironinstitute / jax-finufft
  JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library
  ☆104Updated this week
• dfm / extending-jax
  Extending JAX with custom C++ and CUDA code
  ☆400Updated last year
• dirmeier / sbijax
  Simulation-based inference in JAX
  ☆38Updated 6 months ago
• DifferentiableUniverseInitiative / JaxPM
  JAX-powered Cosmological Particle-Mesh N-body Solver
  ☆56Updated this week