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. We may earn a commission when you buy through links labeled 'Ad' on this page.

• google / objax

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

  View on GitHub
  ☆18Mar 20, 2022Updated 4 years ago
• j-towns / fastar

  View on GitHub
  ☆16Jun 3, 2025Updated 9 months ago
• google-deepmind / dm_pix

  View on GitHub
  PIX is an image processing library in JAX, for JAX.
  ☆434Mar 6, 2025Updated last year
• adamhaber / JaxEnt

  View on GitHub
  Jax-based MaxEnt
  ☆17Nov 24, 2019Updated 6 years ago
• timvieira / rl

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

  View on GitHub
  ☆33Jan 14, 2021Updated 5 years ago
• rlouf / mcx

  View on GitHub
  Express & compile probabilistic programs for performant inference on CPU & GPU. Powered by JAX.
  ☆330Mar 20, 2024Updated 2 years ago
• kzhai / PyInfVoc

  View on GitHub
  An Online Latent Dirichlet Allocation with Infinite Vocabulary implementation in Python.
  ☆12Oct 4, 2018Updated 7 years ago
• harvardnlp / pytorch-struct

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

  View on GitHub
  Monotone operator equilibrium networks
  ☆53Jun 22, 2020Updated 5 years ago
• PhilipVinc / Jax.jl

  View on GitHub
  ☆28Jan 12, 2022Updated 4 years ago
• SymJAX / SymJAX

  View on GitHub
  Documentation:
  ☆130May 22, 2023Updated 2 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,628Mar 25, 2022Updated 3 years ago
• poets-ai / elegy

  View on GitHub
  A High Level API for Deep Learning in JAX
  ☆476Dec 15, 2022Updated 3 years ago
• google-deepmind / dm-haiku

  View on GitHub
  JAX-based neural network library
  ☆3,205Updated this week
• google-deepmind / chex

  View on GitHub
  ☆929Mar 12, 2026Updated last week
• 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
• ericjang / nf-jax

  View on GitHub
  Normalizing Flows in Jax
  ☆109Aug 19, 2020Updated 5 years ago
• cgarciae / treex

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

  View on GitHub
  ☆627Jan 23, 2026Updated 2 months ago
• google-deepmind / rlax

  View on GitHub
  ☆1,413Mar 2, 2026Updated 3 weeks ago
• pyro-ppl / funsor

  View on GitHub
  Functional tensors for probabilistic programming
  ☆245Aug 31, 2023Updated 2 years ago
• cgarciae / einop

  View on GitHub
  ☆60Mar 8, 2022Updated 4 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
• srush / jax-lda

  View on GitHub
  ☆30Feb 11, 2022Updated 4 years ago
• ethanluoyc / dhmc-jax

  View on GitHub
  Discontinuous Hamiltonian Monte Carlo in JAX
  ☆42Feb 24, 2020Updated 6 years ago
• google-deepmind / tensor_annotations

  View on GitHub
  Annotating tensor shapes using Python types
  ☆159Jul 7, 2023Updated 2 years ago
• ericjang / maml-jax

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

  View on GitHub
  Live Python Notebooks with any Editor
  ☆279Dec 7, 2022Updated 3 years ago
• gehring / fax

  View on GitHub
  ☆80Sep 4, 2021Updated 4 years ago
• paperswithcode / axcell

  View on GitHub
  Tools for extracting tables and results from Machine Learning papers
  ☆435Nov 28, 2022Updated 3 years ago
• julien-c / trainer-proposal

  View on GitHub
  ☆13Mar 27, 2020Updated 5 years ago
• google / tree-math

  View on GitHub
  Mathematical operations for JAX pytrees
  ☆210Dec 5, 2024Updated last year
• ikostrikov / pytorch-flows

  View on GitHub
  PyTorch implementations of algorithms for density estimation
  ☆586May 13, 2021Updated 4 years ago
• google / neural-tangents

  View on GitHub
  Fast and Easy Infinite Neural Networks in Python
  ☆2,379Mar 1, 2024Updated 2 years ago
• mattjj / ode-diff-notes

  View on GitHub
  ☆12Oct 8, 2020Updated 5 years ago
• google-deepmind / optax

  View on GitHub
  Optax is a gradient processing and optimization library for JAX.
  ☆2,221Updated this week