Alternatives and similar repositories for jaxnet

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

• PhilipVinc / Jax.jl

  View on GitHub
  ☆28Jan 12, 2022Updated 4 years ago
• FluxML / YaoFlux.jl

  View on GitHub
  Differentiable programming on quantum circuits with Flux
  ☆19Feb 12, 2020Updated 6 years ago
• duvenaud / jaxde

  View on GitHub
  Prototypes of differentiable differential equation solvers in JAX.
  ☆27Feb 3, 2020Updated 6 years ago
• j-towns / fastar

  View on GitHub
  ☆16Jun 3, 2025Updated 8 months ago
• SymJAX / SymJAX

  View on GitHub
  Documentation:
  ☆127May 22, 2023Updated 2 years ago
• gehring / fax

  View on GitHub
  ☆80Sep 4, 2021Updated 4 years ago
• google / tree-math

  View on GitHub
  Mathematical operations for JAX pytrees
  ☆208Dec 5, 2024Updated last year
• ericjang / nf-jax

  View on GitHub
  Normalizing Flows in Jax
  ☆107Aug 19, 2020Updated 5 years ago
• ExpectationMax / sklearn-jax-kernels

  View on GitHub
  Composable kernels for scikit-learn implemented in JAX.
  ☆47Oct 26, 2020Updated 5 years ago
• FluxML / XLA.jl

  View on GitHub
  "Maybe we have our own magic."
  ☆48Jun 17, 2020Updated 5 years ago
• google-deepmind / dm-haiku

  View on GitHub
  JAX-based neural network library
  ☆3,185Jan 30, 2026Updated 2 weeks ago
• n2cholas / shapecheck

  View on GitHub
  Framework-agnostic library for checking array/tensor shapes at runtime.
  ☆46May 4, 2021Updated 4 years ago
• poets-ai / elegy

  View on GitHub
  A High Level API for Deep Learning in JAX
  ☆476Dec 15, 2022Updated 3 years ago
• rtqichen / residual-flows

  View on GitHub
  code for "Residual Flows for Invertible Generative Modeling".
  ☆273Sep 19, 2023Updated 2 years ago
• google / neural-tangents

  View on GitHub
  Fast and Easy Infinite Neural Networks in Python
  ☆2,374Mar 1, 2024Updated last year
• 4rtemi5 / imax

  View on GitHub
  Image augmentation library for Jax
  ☆41Apr 9, 2024Updated last year
• PrincetonLIPS / randomized_telescopes

  View on GitHub
  Code for "Efficient optimization of loops and limits with randomized telescoping sums"
  ☆27May 13, 2019Updated 6 years ago
• google-deepmind / dm_pix

  View on GitHub
  PIX is an image processing library in JAX, for JAX.
  ☆433Mar 6, 2025Updated 11 months ago
• chaserileyroberts / TensorNetsForML

  View on GitHub
  Code for the paper "Tensor Networks for Maching Learning"
  ☆17Nov 7, 2019Updated 6 years ago
• dot-delta / Meta-SGD

  View on GitHub
  Meta-SGD Algorithms Implementation
  ☆22Jul 18, 2024Updated last year
• rlouf / blog-benchmark-rwmetropolis

  View on GitHub
  Code to reproduce the results of 👇
  ☆18Aug 19, 2022Updated 3 years ago
• google-deepmind / optax

  View on GitHub
  Optax is a gradient processing and optimization library for JAX.
  ☆2,177Updated this week
• microsoft / coax

  View on GitHub
  This project was moved to: https://github.com/coax-dev/coax
  ☆161Nov 28, 2022Updated 3 years ago
• google-deepmind / rlax

  View on GitHub
  ☆1,404Dec 9, 2025Updated 2 months ago
• Information-Fusion-Lab-Umass / NuX

  View on GitHub
  Normalizing Flows using JAX
  ☆86Nov 30, 2023Updated 2 years ago
• srush / parallax

  View on GitHub
  ☆153May 25, 2020Updated 5 years ago
• Santosh-Gupta / SpeedTorch

  View on GitHub
  Library for faster pinned CPU <-> GPU transfer in Pytorch
  ☆683Feb 21, 2020Updated 5 years ago
• dfm / extending-jax

  View on GitHub
  Extending JAX with custom C++ and CUDA code
  ☆403Aug 18, 2024Updated last year
• buwantaiji / FermiFlow

  View on GitHub
  Ab-initio simulation of interacting fermions with equivariant normalizing flow.
  ☆34Oct 23, 2022Updated 3 years ago
• wulfebw / async_rl

  View on GitHub
  Python implementation of tabular asynchronous actor critic
  ☆11May 3, 2016Updated 9 years ago
• Lifelong-ML / LASEM

  View on GitHub
  Code for the ICML 2021 paper "Sharing Less is More: Lifelong Learning in Deep Networks with Selective Layer Transfer"
  ☆12Aug 17, 2021Updated 4 years ago
• google-research / computation-thru-dynamics

  View on GitHub
  Understanding computation in artificial and biological recurrent networks through the lens of dynamical systems.
  ☆390May 28, 2021Updated 4 years ago
• google / jaxopt

  View on GitHub
  Hardware accelerated, batchable and differentiable optimizers in JAX.
  ☆1,026Dec 17, 2025Updated last month
• QuantumBFS / QuAlgorithmZoo.jl

  View on GitHub
  A curated implementation of quantum algorithms with Yao.jl
  ☆107Aug 21, 2023Updated 2 years ago
• BlackHC / tfpyth

  View on GitHub
  Putting TensorFlow back in PyTorch, back in TensorFlow (differentiable TensorFlow PyTorch adapters).
  ☆647Nov 30, 2020Updated 5 years ago
• revbucket / geometric-certificates

  View on GitHub
  Geometric Certifications of Neural Nets
  ☆42Nov 22, 2022Updated 3 years ago
• rserizel / minibatchNMF

  View on GitHub
  Mini-batch multiplicative updates for beta-NMF
  ☆15Jun 27, 2017Updated 8 years ago
• djsutherland / igms

  View on GitHub
  Implicit generative models and related stuff based on the MMD, in PyTorch
  ☆16Sep 24, 2020Updated 5 years ago
• musyoku / chainer-qrnn

  View on GitHub
  Chainer implementation of Q-RNN (Quasi-Reccurent Neural Networks)
  ☆12Aug 17, 2017Updated 8 years ago