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
• jxbz / fromage
  🧀 Pytorch code for the Fromage optimiser.
  ☆124Updated 10 months ago
• hardmaru / mdn_jax_tutorial
  Mixture Density Networks (Bishop, 1994) tutorial in JAX
  ☆59Updated 5 years ago
• ofnote / tsalib
  Tensor Shape Annotation Library (numpy, tensorflow, pytorch, ...)
  ☆266Updated 5 years ago
• srush / awesome-ml-tracking
  ☆103Updated 4 years ago
• srush / torch-queue
  ☆67Updated last year
• ericjang / nf-jax
  Normalizing Flows in Jax
  ☆108Updated 4 years ago
• SymJAX / SymJAX
  Documentation:
  ☆120Updated 2 years ago
• jxbz / madam
  👩 Pytorch and Jax code for the Madam optimiser.
  ☆51Updated 4 years ago
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆110Updated 2 years ago
• google-deepmind / jaxline
  ☆157Updated last year
• g-benton / loss-surface-simplexes
  ☆99Updated 3 years ago
• aiwabdn / pygln
  Python implementation of GLN in different frameworks
  ☆98Updated 4 years ago
• cybertronai / ncluster
  Lightweight interface to AWS
  ☆47Updated 5 years ago
• n2cholas / shapecheck
  Framework-agnostic library for checking array/tensor shapes at runtime.
  ☆46Updated 4 years ago
• AndreasMadsen / stable-nalu
  Code for Neural Arithmetic Units (ICLR) and Measuring Arithmetic Extrapolation Performance (SEDL|NeurIPS)
  ☆146Updated 3 years ago
• kyunghyuncho / jax-practice
  ☆12Updated 4 years ago
• pyro-ppl / funsor
  Functional tensors for probabilistic programming
  ☆241Updated last year
• facebookresearch / dagger
  Experiment orchestration
  ☆103Updated 4 years ago
• clemkoa / ntm
  Neural Turing Machines in pytorch
  ☆48Updated 3 years ago
• google-deepmind / einshape
  ☆104Updated 11 months ago
• namedtensor / notation
  ☆108Updated 2 years ago
• harvardnlp / namedtensor
  Named Tensor implementation for Torch
  ☆444Updated 2 years ago
• viking-sudo-rm / stacknn-core
  Pip-installable differentiable stacks in PyTorch!
  ☆65Updated 4 years ago
• SirRob1997 / Crowded-Valley---Results
  This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"
  ☆180Updated 3 years ago
• uclnlp / torch-imle
  Implicit MLE: Backpropagating Through Discrete Exponential Family Distributions
  ☆258Updated last year
• szymonmaszke / torchfunc
  PyTorch functions and utilities to make your life easier
  ☆195Updated 4 years ago
• xflash96 / pybind11_package_example
  Example python package with pybind11 cpp extension
  ☆57Updated 4 years ago
• cybertronai / transformer-xl
  Training Transformer-XL on 128 GPUs
  ☆140Updated 4 years ago
• devzhk / Implicit-Competitive-Regularization
  Code for: Implicit Competitive Regularization in GANs
  ☆113Updated 3 years ago