Alternatives and similar repositories for parallax:

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

• cgarciae / treex
  A Pytree Module system for Deep Learning in JAX
  ☆213Updated last year
• hardmaru / mdn_jax_tutorial
  Mixture Density Networks (Bishop, 1994) tutorial in JAX
  ☆58Updated 4 years ago
• jxbz / fromage
  🧀 Pytorch code for the Fromage optimiser.
  ☆123Updated 7 months ago
• jxbz / madam
  👩 Pytorch and Jax code for the Madam optimiser.
  ☆51Updated 4 years ago
• SymJAX / SymJAX
  Documentation:
  ☆119Updated last year
• n2cholas / shapecheck
  Framework-agnostic library for checking array/tensor shapes at runtime.
  ☆47Updated 3 years ago
• aiwabdn / pygln
  Python implementation of GLN in different frameworks
  ☆98Updated 4 years ago
• ericjang / nf-jax
  Normalizing Flows in Jax
  ☆107Updated 4 years ago
• ofnote / tsalib
  Tensor Shape Annotation Library (numpy, tensorflow, pytorch, ...)
  ☆264Updated 4 years ago
• cybertronai / ncluster
  Lightweight interface to AWS
  ☆47Updated 5 years ago
• facebookresearch / dagger
  Experiment orchestration
  ☆103Updated 4 years ago
• google-deepmind / jaxline
  ☆153Updated last year
• srush / torch-queue
  ☆67Updated last year
• viking-sudo-rm / stacknn-core
  Pip-installable differentiable stacks in PyTorch!
  ☆65Updated 4 years ago
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆107Updated 2 years ago
• PetrochukM / HParams
  Configure Python functions explicitly and safely
  ☆126Updated 3 months ago
• namedtensor / notation
  ☆108Updated 2 years ago
• rolandgvc / flaxvision
  A selection of neural network models ported from torchvision for JAX & Flax.
  ☆44Updated 4 years ago
• johnhw / differentiable_sorting
  Differentiable bitonic sorting
  ☆138Updated 4 years ago
• willwhitney / reprieve
  A library for evaluating representations.
  ☆74Updated 3 years ago
• mcleonard / xaby
  Functional machine learning for fun
  ☆84Updated 3 years ago
• devzhk / Implicit-Competitive-Regularization
  Code for: Implicit Competitive Regularization in GANs
  ☆114Updated 3 years ago
• mblondel / fenchel-young-losses
  Probabilistic classification in PyTorch/TensorFlow/scikit-learn with Fenchel-Young losses
  ☆183Updated last year
• clemkoa / ntm
  Neural Turing Machines in pytorch
  ☆48Updated 3 years ago
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆57Updated last year
• conormdurkan / autoregressive-energy-machines
  Autoregressive Energy Machines
  ☆77Updated 2 years ago
• srush / awesome-ml-tracking
  ☆102Updated 4 years ago
• pyro-ppl / funsor
  Functional tensors for probabilistic programming
  ☆237Updated last year
• g-benton / loss-surface-simplexes
  ☆98Updated 3 years ago
• minitorch / minitorch.github.io
  Docs
  ☆143Updated 2 months ago