Alternatives and similar repositories for jax_sourceror

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

• j-towns / scanagram

  View on GitHub
  Tidy autoregressive inference in JAX
  ☆15Sep 1, 2025Updated 5 months ago
• vboussange / jaxscape

  View on GitHub
  A minimal JAX library for connectivity modelling at scale
  ☆11Jan 27, 2026Updated 3 weeks ago
• davisyoshida / lorax

  View on GitHub
  LoRA for arbitrary JAX models and functions
  ☆145Feb 26, 2024Updated last year
• alonfnt / pcax

  View on GitHub
  Differentiable Principal Component Analysis (PCA) implementation in JAX
  ☆32Apr 19, 2025Updated 9 months ago
• maxwells-daemons / deltaflow

  View on GitHub
  A GPU-accelerated differentiable fluid simulator written in JAX.
  ☆11Feb 1, 2021Updated 5 years ago
• fidel-schaposnik / muzero

  View on GitHub
  Tensorflow implementation of MuZero algorithm
  ☆11Aug 23, 2022Updated 3 years ago
• kelechi-c / dit_flow

  View on GitHub
  DiT (training + flow matching) in Jax
  ☆11Jan 5, 2025Updated last year
• fattorib / transformer_shmap

  View on GitHub
  Tensor Parallelism with JAX + Shard Map
  ☆11Sep 29, 2023Updated 2 years ago
• y1xiaoc / fwdlap

  View on GitHub
  Forward mode laplacian implemented in JAX tracer
  ☆30Jan 7, 2026Updated last month
• nestordemeure / jochastic

  View on GitHub
  A JAX implementation of stochastic addition.
  ☆14Aug 15, 2022Updated 3 years ago
• opooladz / Preconditioned-Stochastic-Gradient-Descent

  View on GitHub
  A repo based on XiLin Li's PSGD repo that extends some of the experiments.
  ☆14Oct 7, 2024Updated last year
• ituvisionlab / EdVAE

  View on GitHub
  Official PyTorch implementation of "EdVAE: Mitigating Codebook Collapse with Evidential Discrete Variational Autoencoders"
  ☆14Sep 20, 2024Updated last year
• diffprog / diffprog.github.io

  View on GitHub
  Website for the book "The Elements of Differentiable Programming".
  ☆14Jul 2, 2025Updated 7 months ago
• JAXtronomy / spexial

  View on GitHub
  Special functions in Jax
  ☆16Mar 1, 2025Updated 11 months ago
• young-geng / tpu_pod_commander

  View on GitHub
  TPU pod commander is a package for managing and launching jobs on Google Cloud TPU pods.
  ☆21Sep 24, 2025Updated 4 months ago
• anh-tong / fractional_neural_sde

  View on GitHub
  Fractional White Noises for Neural Stochastic Differential Equations (NeurIPS 2022)
  ☆16Nov 17, 2022Updated 3 years ago
• GallagherCommaJack / modulax

  View on GitHub
  ☆18Aug 24, 2024Updated last year
• timeolord / Reinforcement-Learning-Stock-Trader

  View on GitHub
  Using a modified version of Werner Duvaud's MuZero implementation (https://github.com/werner-duvaud/muzero-general) this reinforcement ag…
  ☆18Jun 30, 2021Updated 4 years ago
• xl0 / lovely-jax

  View on GitHub
  JAX Arrays for human consumption
  ☆111Oct 23, 2025Updated 3 months ago
• kerry-he / qics

  View on GitHub
  Conic solver for optimization problems in quantum information theory
  ☆26Apr 20, 2025Updated 9 months ago
• Ceyron / pdequinox

  View on GitHub
  Neural Emulator Architectures in JAX.
  ☆23Nov 5, 2024Updated last year
• thecharlieblake / lovely-llama

  View on GitHub
  An implementation of the Llama architecture, to instruct and delight
  ☆21May 31, 2025Updated 8 months ago
• zchuning / repo

  View on GitHub
  Resilient Model-Based RL by Regularizing Posterior Predictability
  ☆22Mar 4, 2024Updated last year
• berkegokmen1 / slurm-ssh-compute-node

  View on GitHub
  Connect to Slurm Compute Nodes Directly from VSCode. Do your debugging and development easily.
  ☆28Oct 8, 2024Updated last year
• marin-community / haliax

  View on GitHub
  Named Tensors for Legible Deep Learning in JAX
  ☆218Nov 8, 2025Updated 3 months ago
• tum-pbs / diffSPH

  View on GitHub
  ☆43Updated this week
• geraschenko / gmrfs

  View on GitHub
  Gaussian Markov Random Fields (GMRFs) and Integrated Nested Laplace Approximation (INLA)
  ☆22Apr 17, 2024Updated last year
• f0uriest / orthax

  View on GitHub
  Orthogonal polynomials with JAX
  ☆27Jan 20, 2026Updated 3 weeks ago
• flatironinstitute / jax-finufft

  View on GitHub
  JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library
  ☆115Feb 3, 2026Updated 2 weeks ago
• dmusekamp / al4pde

  View on GitHub
  AL4PDE: A Benchmark for Active Learning for Neural PDE Solvers
  ☆29May 30, 2025Updated 8 months ago
• jax-ml / jax-llm-examples

  View on GitHub
  Minimal yet performant LLM examples in pure JAX
  ☆242Jan 14, 2026Updated last month
• qobi / AD-Rosetta-Stone

  View on GitHub
  Examples of Automatic Differentiation (AD) in many different languages and systems
  ☆27Jun 25, 2018Updated 7 years ago
• EEA-sensors / parallel-non-linear-gaussian-smoothers

  View on GitHub
  Companion code in JAX for the paper Parallel Iterated Extended and Sigma-Point Kalman Smoothers.
  ☆27Aug 9, 2024Updated last year
• jax-ml / oryx

  View on GitHub
  Oryx is a library for probabilistic programming and deep learning built on top of Jax.
  ☆298Jan 9, 2026Updated last month
• ucl-bug / jaxdf

  View on GitHub
  A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
  ☆132Feb 9, 2026Updated last week
• google-research / autoconj

  View on GitHub
  Recognizing and exploiting conjugacy without a domain-specific language
  ☆36Dec 1, 2019Updated 6 years ago
• young-geng / mintext

  View on GitHub
  Minimal but scalable implementation of large language models in JAX
  ☆35Nov 28, 2025Updated 2 months ago
• jax-ml / jax-triton

  View on GitHub
  jax-triton contains integrations between JAX and OpenAI Triton
  ☆439Feb 9, 2026Updated last week
• JimOhman / model-based-rl

  View on GitHub
  Implementation of MuZero with PyTorch, based on the pseudocode from DeepMind (https://arxiv.org/src/1911.08265v2/anc/pseudocode.py).
  ☆33Aug 14, 2022Updated 3 years ago