srush/parallax external links

---

GitHub - View repository

GitHub.dev - Open in online code editor

Star history - Growth chart over time

deps.dev - Dependencies and security vulnerabilities

Gitingest - Export source code for LLM usage

Repomix - Export source code for LLM usage

uithub - Export source code for LLM usage

Close

srush/parallax

Readme badge preview -

If you own this repo, copy the snippet below and add it to your README.md

---

Copy

[![RelatedRepos](https://img.shields.io/badge/related-repos-yellow)](https://relatedrepos.com/gh/srush/parallax)

Close

srush / parallaxView on GitHubMore

• External links
• Readme badge

☆153May 25, 2020Updated 5 years ago

Alternatives and similar repositories for parallax

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

• google / objax

  View on GitHub
  ☆774Jan 27, 2024Updated 2 years ago
• harvardnlp / strux

  View on GitHub
  ☆18Mar 20, 2022Updated 3 years ago
• google-deepmind / dm_pix

  View on GitHub
  PIX is an image processing library in JAX, for JAX.
  ☆434Mar 6, 2025Updated 11 months ago
• srush / learns-dex

  View on GitHub
  ☆33Jan 14, 2021Updated 5 years ago
• timvieira / rl

  View on GitHub
  Reference implementation of algorithms for reinforcement learning and Markov decision processes.
  ☆12Jan 28, 2021Updated 5 years ago
• j-towns / fastar

  View on GitHub
  ☆16Jun 3, 2025Updated 9 months ago
• rlouf / mcx

  View on GitHub
  Express & compile probabilistic programs for performant inference on CPU & GPU. Powered by JAX.
  ☆330Mar 20, 2024Updated last year
• locuslab / monotone_op_net

  View on GitHub
  Monotone operator equilibrium networks
  ☆53Jun 22, 2020Updated 5 years ago
• facebookresearch / higher

  View on GitHub
  higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual tr…
  ☆1,627Mar 25, 2022Updated 3 years ago
• adamhaber / JaxEnt

  View on GitHub
  Jax-based MaxEnt
  ☆17Nov 24, 2019Updated 6 years ago
• harvardnlp / pytorch-struct

  View on GitHub
  Fast, general, and tested differentiable structured prediction in PyTorch
  ☆1,123Apr 20, 2022Updated 3 years ago
• PrincetonLIPS / randomized_telescopes

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

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

  View on GitHub
  ☆30Feb 11, 2022Updated 4 years ago
• google-deepmind / chex

  View on GitHub
  ☆925Jan 29, 2026Updated last month
• google-deepmind / dm-haiku

  View on GitHub
  JAX-based neural network library
  ☆3,191Feb 24, 2026Updated last week
• PhilipVinc / Jax.jl

  View on GitHub
  ☆28Jan 12, 2022Updated 4 years ago
• google / tree-math

  View on GitHub
  Mathematical operations for JAX pytrees
  ☆209Dec 5, 2024Updated last year
• srush / streambook

  View on GitHub
  Live Python Notebooks with any Editor
  ☆279Dec 7, 2022Updated 3 years ago
• google-deepmind / rlax

  View on GitHub
  ☆1,407Dec 9, 2025Updated 2 months ago
• poets-ai / elegy

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

  View on GitHub
  Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX
  ☆103Jul 6, 2023Updated 2 years ago
• cgarciae / treex

  View on GitHub
  A Pytree Module system for Deep Learning in JAX
  ☆213Feb 26, 2023Updated 3 years ago
• google-deepmind / distrax

  View on GitHub
  ☆620Jan 23, 2026Updated last month
• SymJAX / SymJAX

  View on GitHub
  Documentation:
  ☆129May 22, 2023Updated 2 years ago
• ericjang / nf-jax

  View on GitHub
  Normalizing Flows in Jax
  ☆107Aug 19, 2020Updated 5 years ago
• pyro-ppl / funsor

  View on GitHub
  Functional tensors for probabilistic programming
  ☆245Aug 31, 2023Updated 2 years ago
• ikostrikov / pytorch-flows

  View on GitHub
  PyTorch implementations of algorithms for density estimation
  ☆586May 13, 2021Updated 4 years ago
• cgarciae / einop

  View on GitHub
  ☆60Mar 8, 2022Updated 3 years ago
• ericjang / maml-jax

  View on GitHub
  Implementation of Model-Agnostic Meta-Learning (MAML) in Jax
  ☆191Sep 28, 2022Updated 3 years ago
• noahgolmant / pytorch-hessian-eigenthings

  View on GitHub
  Efficient PyTorch Hessian eigendecomposition tools!
  ☆386Feb 29, 2024Updated 2 years ago
• google / neural-tangents

  View on GitHub
  Fast and Easy Infinite Neural Networks in Python
  ☆2,375Mar 1, 2024Updated 2 years ago
• yobibyte / pgn

  View on GitHub
  Graph Nets in pytorch
  ☆27Dec 8, 2022Updated 3 years ago
• google-research / rigl

  View on GitHub
  End-to-end training of sparse deep neural networks with little-to-no performance loss.
  ☆335Jan 26, 2023Updated 3 years ago
• osqp / osqpth

  View on GitHub
  The differentiable OSQP solver layer for PyTorch
  ☆65Dec 26, 2022Updated 3 years ago
• taufikxu / FD-ScoreMatching

  View on GitHub
  The official code for Efficient Learning of Generative Models via Finite-Difference Score Matching
  ☆12Nov 22, 2022Updated 3 years ago
• paperswithcode / axcell

  View on GitHub
  Tools for extracting tables and results from Machine Learning papers
  ☆435Nov 28, 2022Updated 3 years ago
• cgarciae / ciclo

  View on GitHub
  A functional training loops library for JAX
  ☆88Feb 13, 2024Updated 2 years ago
• gehring / fax

  View on GitHub
  ☆80Sep 4, 2021Updated 4 years ago