Alternatives and similar repositories for parallax

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

• hardmaru / mdn_jax_tutorial
  Mixture Density Networks (Bishop, 1994) tutorial in JAX
  ☆61Updated 5 years ago
• n2cholas / shapecheck
  Framework-agnostic library for checking array/tensor shapes at runtime.
  ☆46Updated 4 years ago
• ofnote / tsalib
  Tensor Shape Annotation Library (numpy, tensorflow, pytorch, ...)
  ☆266Updated 5 years ago
• srush / torch-queue
  ☆71Updated 2 years ago
• cgarciae / treex
  A Pytree Module system for Deep Learning in JAX
  ☆214Updated 2 years ago
• jxbz / fromage
  🧀 Pytorch code for the Fromage optimiser.
  ☆129Updated last year
• aiwabdn / pygln
  Python implementation of GLN in different frameworks
  ☆97Updated 5 years ago
• google-deepmind / jaxline
  ☆161Updated last year
• jxbz / madam
  👩 Pytorch and Jax code for the Madam optimiser.
  ☆53Updated 4 years ago
• viking-sudo-rm / stacknn-core
  Pip-installable differentiable stacks in PyTorch!
  ☆65Updated 5 years ago
• ericjang / maml-jax
  Implementation of Model-Agnostic Meta-Learning (MAML) in Jax
  ☆191Updated 3 years ago
• uclnlp / torch-imle
  Implicit MLE: Backpropagating Through Discrete Exponential Family Distributions
  ☆259Updated 2 years ago
• kyunghyuncho / jax-practice
  ☆13Updated 5 years ago
• clemkoa / ntm
  Neural Turing Machines in pytorch
  ☆48Updated 3 years ago
• g-benton / loss-surface-simplexes
  ☆100Updated 3 years ago
• facebookresearch / dagger
  Experiment orchestration
  ☆102Updated 5 years ago
• minitorch / minitorch.github.io
  Docs
  ☆143Updated last year
• cybertronai / ncluster
  Lightweight interface to AWS
  ☆47Updated 6 years ago
• rolandgvc / flaxvision
  A selection of neural network models ported from torchvision for JAX & Flax.
  ☆45Updated 4 months ago
• ericjang / nf-jax
  Normalizing Flows in Jax
  ☆107Updated 5 years ago
• thegregyang / GP4A
  Code for NeurIPS 2019 paper: "Tensor Programs I: Wide Feedforward or Recurrent Neural Networks of Any Architecture are Gaussian Processes…
  ☆248Updated 5 years ago
• willwhitney / reprieve
  A library for evaluating representations.
  ☆77Updated 4 years ago
• mle-infrastructure / mle-scheduler
  Lightweight Cluster/Cloud VM Job Management 🚀
  ☆42Updated last year
• PetrochukM / HParams
  Configure Python functions explicitly and safely
  ☆128Updated last year
• srush / awesome-ml-tracking
  ☆104Updated 4 years ago
• universome / loss-patterns
  Loss Patterns of Neural Networks
  ☆86Updated 4 years ago
• AndreasMadsen / stable-nalu
  Code for Neural Arithmetic Units (ICLR) and Measuring Arithmetic Extrapolation Performance (SEDL|NeurIPS)
  ☆146Updated 4 years ago
• johnhw / differentiable_sorting
  Differentiable bitonic sorting
  ☆141Updated 5 years ago
• kingoflolz / swarm-jax
  Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes
  ☆242Updated 2 years ago
• IssamLaradji / sps
  Official code for the Stochastic Polyak step-size optimizer
  ☆139Updated last year