Alternatives and similar repositories for parallax:

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

• 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
• cgarciae / treex
  A Pytree Module system for Deep Learning in JAX
  ☆213Updated 2 years ago
• n2cholas / shapecheck
  Framework-agnostic library for checking array/tensor shapes at runtime.
  ☆47Updated 3 years ago
• hardmaru / mdn_jax_tutorial
  Mixture Density Networks (Bishop, 1994) tutorial in JAX
  ☆58Updated 4 years ago
• ofnote / tsalib
  Tensor Shape Annotation Library (numpy, tensorflow, pytorch, ...)
  ☆264Updated 4 years ago
• SymJAX / SymJAX
  Documentation:
  ☆119Updated last year
• willwhitney / reprieve
  A library for evaluating representations.
  ☆75Updated 3 years ago
• g-benton / loss-surface-simplexes
  ☆98Updated 3 years ago
• google-deepmind / jaxline
  ☆154Updated last year
• srush / torch-queue
  ☆68Updated last year
• aiwabdn / pygln
  Python implementation of GLN in different frameworks
  ☆98Updated 4 years ago
• devzhk / Implicit-Competitive-Regularization
  Code for: Implicit Competitive Regularization in GANs
  ☆114Updated 3 years ago
• srush / awesome-ml-tracking
  ☆102Updated 4 years ago
• rolandgvc / flaxvision
  A selection of neural network models ported from torchvision for JAX & Flax.
  ☆44Updated 4 years ago
• ericjang / nf-jax
  Normalizing Flows in Jax
  ☆107Updated 4 years ago
• facebookresearch / dagger
  Experiment orchestration
  ☆103Updated 4 years ago
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆108Updated 2 years ago
• namedtensor / notation
  ☆108Updated 2 years ago
• viking-sudo-rm / stacknn-core
  Pip-installable differentiable stacks in PyTorch!
  ☆65Updated 4 years ago
• pyro-ppl / funsor
  Functional tensors for probabilistic programming
  ☆238Updated last year
• kyunghyuncho / jax-practice
  ☆12Updated 4 years ago
• google-deepmind / tensor_annotations
  Annotating tensor shapes using Python types
  ☆158Updated last year
• mcleonard / xaby
  Functional machine learning for fun
  ☆84Updated 3 years ago
• clemkoa / ntm
  Neural Turing Machines in pytorch
  ☆48Updated 3 years ago
• JarnoRFB / incense
  Interactively retrieve data from sacred experiments.
  ☆82Updated this week
• cybertronai / transformer-xl
  Training Transformer-XL on 128 GPUs
  ☆140Updated 4 years ago
• y0ast / slurm-for-ml
  A Machine Learning workflow for Slurm.
  ☆149Updated 4 years ago
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆38Updated 11 months ago
• mle-infrastructure / mle-scheduler
  Lightweight Cluster/Cloud VM Job Management 🚀
  ☆40Updated 6 months ago