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
  ☆214Updated 2 years ago
• hardmaru / mdn_jax_tutorial
  Mixture Density Networks (Bishop, 1994) tutorial in JAX
  ☆59Updated 4 years ago
• n2cholas / shapecheck
  Framework-agnostic library for checking array/tensor shapes at runtime.
  ☆46Updated 3 years ago
• SymJAX / SymJAX
  Documentation:
  ☆119Updated last year
• jxbz / fromage
  🧀 Pytorch code for the Fromage optimiser.
  ☆123Updated 8 months ago
• google-deepmind / jaxline
  ☆154Updated last year
• aiwabdn / pygln
  Python implementation of GLN in different frameworks
  ☆98Updated 4 years ago
• srush / torch-queue
  ☆67Updated last year
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆108Updated 2 years ago
• ericjang / nf-jax
  Normalizing Flows in Jax
  ☆107Updated 4 years ago
• g-benton / loss-surface-simplexes
  ☆99Updated 3 years ago
• devzhk / Implicit-Competitive-Regularization
  Code for: Implicit Competitive Regularization in GANs
  ☆114Updated 3 years ago
• szymonmaszke / torchfunc
  PyTorch functions and utilities to make your life easier
  ☆195Updated 4 years ago
• mblondel / fenchel-young-losses
  Probabilistic classification in PyTorch/TensorFlow/scikit-learn with Fenchel-Young losses
  ☆184Updated last year
• rolandgvc / flaxvision
  A selection of neural network models ported from torchvision for JAX & Flax.
  ☆44Updated 4 years ago
• clemkoa / ntm
  Neural Turing Machines in pytorch
  ☆48Updated 3 years ago
• viking-sudo-rm / stacknn-core
  Pip-installable differentiable stacks in PyTorch!
  ☆65Updated 4 years ago
• ofnote / tsalib
  Tensor Shape Annotation Library (numpy, tensorflow, pytorch, ...)
  ☆265Updated 4 years ago
• pyro-ppl / funsor
  Functional tensors for probabilistic programming
  ☆239Updated last year
• jxbz / madam
  👩 Pytorch and Jax code for the Madam optimiser.
  ☆51Updated 4 years ago
• facebookresearch / dagger
  Experiment orchestration
  ☆103Updated 4 years ago
• johnhw / differentiable_sorting
  Differentiable bitonic sorting
  ☆139Updated 4 years ago
• google-deepmind / einshape
  ☆101Updated 9 months ago
• namedtensor / notation
  ☆108Updated 2 years ago
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆56Updated 2 years ago
• kyunghyuncho / jax-practice
  ☆12Updated 4 years ago
• y0ast / slurm-for-ml
  A Machine Learning workflow for Slurm.
  ☆149Updated 4 years ago
• willwhitney / reprieve
  A library for evaluating representations.
  ☆76Updated 3 years ago
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆39Updated last year
• PrincetonLIPS / RandomizedAutomaticDifferentiation
  Experiment code for "Randomized Automatic Differentiation"
  ☆66Updated 4 years ago