srush / parallax

Related projects ⓘ

Alternatives and complementary repositories for parallax

• cgarciae / treex
  A Pytree Module system for Deep Learning in JAX
  ☆214Updated last year
• srush / torch-queue
  ☆67Updated last year
• hardmaru / mdn_jax_tutorial
  Mixture Density Networks (Bishop, 1994) tutorial in JAX
  ☆58Updated 4 years ago
• SymJAX / SymJAX
  Documentation:
  ☆120Updated last year
• jxbz / fromage
  🧀 Pytorch code for the Fromage optimiser.
  ☆122Updated 4 months ago
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆104Updated 2 years ago
• jxbz / madam
  👩 Pytorch and Jax code for the Madam optimiser.
  ☆51Updated 3 years ago
• aiwabdn / pygln
  Python implementation of GLN in different frameworks
  ☆95Updated 4 years ago
• n2cholas / shapecheck
  Framework-agnostic library for checking array/tensor shapes at runtime.
  ☆47Updated 3 years ago
• facebookresearch / dagger
  Experiment orchestration
  ☆103Updated 4 years ago
• ofnote / tsalib
  Tensor Shape Annotation Library (numpy, tensorflow, pytorch, ...)
  ☆262Updated 4 years ago
• google-deepmind / jaxline
  ☆153Updated 11 months ago
• viking-sudo-rm / stacknn-core
  Pip-installable differentiable stacks in PyTorch!
  ☆65Updated 4 years ago
• willwhitney / reprieve
  A library for evaluating representations.
  ☆76Updated 2 years ago
• devzhk / Implicit-Competitive-Regularization
  Code for: Implicit Competitive Regularization in GANs
  ☆114Updated 2 years ago
• ericjang / nf-jax
  Normalizing Flows in Jax
  ☆105Updated 4 years ago
• cybertronai / transformer-xl
  Training Transformer-XL on 128 GPUs
  ☆140Updated 4 years ago
• srush / awesome-ml-tracking
  ☆101Updated 3 years ago
• cybertronai / ncluster
  Lightweight interface to AWS
  ☆47Updated 5 years ago
• j-towns / vdvae-jax
  Very deep VAEs in JAX/Flax
  ☆45Updated 3 years ago
• fsschneider / DeepOBS
  DeepOBS: A Deep Learning Optimizer Benchmark Suite
  ☆103Updated 11 months ago
• pyro-ppl / funsor
  Functional tensors for probabilistic programming
  ☆236Updated last year
• kingoflolz / swarm-jax
  Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes
  ☆237Updated last year
• rolandgvc / flaxvision
  A selection of neural network models ported from torchvision for JAX & Flax.
  ☆44Updated 3 years ago
• namedtensor / notation
  ☆108Updated last year
• hal3 / macarico
  learning to search in pytorch
  ☆111Updated 4 years ago
• mblondel / fenchel-young-losses
  Probabilistic classification in PyTorch/TensorFlow/scikit-learn with Fenchel-Young losses
  ☆183Updated last year
• szymonmaszke / torchfunc
  PyTorch functions and utilities to make your life easier
  ☆195Updated 3 years ago
• xflash96 / pybind11_package_example
  Example python package with pybind11 cpp extension
  ☆57Updated 3 years ago
• ericjang / maml-jax
  Implementation of Model-Agnostic Meta-Learning (MAML) in Jax
  ☆188Updated 2 years ago