srush / parallax

Related projects ⓘ

Alternatives and complementary repositories for parallax

• cgarciae / treex
  A Pytree Module system for Deep Learning in JAX
  ☆214Updated last year
• ofnote / tsalib
  Tensor Shape Annotation Library (numpy, tensorflow, pytorch, ...)
  ☆261Updated 4 years ago
• jxbz / madam
  👩 Pytorch and Jax code for the Madam optimiser.
  ☆50Updated 3 years ago
• hardmaru / mdn_jax_tutorial
  Mixture Density Networks (Bishop, 1994) tutorial in JAX
  ☆58Updated 4 years ago
• SymJAX / SymJAX
  Documentation:
  ☆120Updated last year
• ericjang / nf-jax
  Normalizing Flows in Jax
  ☆105Updated 4 years ago
• n2cholas / shapecheck
  Framework-agnostic library for checking array/tensor shapes at runtime.
  ☆47Updated 3 years ago
• google-deepmind / jaxline
  ☆152Updated 10 months ago
• srush / torch-queue
  ☆66Updated last year
• jxbz / fromage
  🧀 Pytorch code for the Fromage optimiser.
  ☆122Updated 3 months ago
• aiwabdn / pygln
  Python implementation of GLN in different frameworks
  ☆95Updated 4 years ago
• viking-sudo-rm / stacknn-core
  Pip-installable differentiable stacks in PyTorch!
  ☆65Updated 4 years ago
• namedtensor / notation
  ☆108Updated last year
• pyro-ppl / funsor
  Functional tensors for probabilistic programming
  ☆236Updated last year
• facebookresearch / dagger
  Experiment orchestration
  ☆103Updated 4 years ago
• willwhitney / reprieve
  A library for evaluating representations.
  ☆76Updated 2 years ago
• g-benton / loss-surface-simplexes
  ☆97Updated 2 years ago
• JarnoRFB / incense
  Interactively retrieve data from sacred experiments.
  ☆81Updated last week
• srush / awesome-ml-tracking
  ☆101Updated 3 years ago
• google-deepmind / tensor_annotations
  Annotating tensor shapes using Python types
  ☆158Updated last year
• kyunghyuncho / jax-practice
  ☆11Updated 4 years ago
• y0ast / slurm-for-ml
  A Machine Learning workflow for Slurm.
  ☆146Updated 3 years ago
• szymonmaszke / torchfunc
  PyTorch functions and utilities to make your life easier
  ☆195Updated 3 years ago
• mle-infrastructure / mle-scheduler
  Lightweight Cluster/Cloud VM Job Management 🚀
  ☆41Updated 2 months ago
• hal3 / macarico
  learning to search in pytorch
  ☆111Updated 4 years ago
• devzhk / Implicit-Competitive-Regularization
  Code for: Implicit Competitive Regularization in GANs
  ☆114Updated 2 years ago
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆104Updated 2 years ago
• cybertronai / ncluster
  Lightweight interface to AWS
  ☆47Updated 5 years ago
• google-deepmind / einshape
  ☆98Updated 4 months ago
• rolandgvc / flaxvision
  A selection of neural network models ported from torchvision for JAX & Flax.
  ☆44Updated 3 years ago