teddykoker / learning-to-learn-jax

Related projects ⓘ

Alternatives and complementary repositories for learning-to-learn-jax

• lucidrains / all-normalization-transformer
  A simple Transformer where the softmax has been replaced with normalization
  ☆18Updated 4 years ago
• lucidrains / deep-linear-network
  A simple implementation of a deep linear Pytorch module
  ☆18Updated 4 years ago
• facebookresearch / grounding-inductive-biases
  reproduces experiments from "Grounding inductive biases in natural images: invariance stems from variations in data"
  ☆16Updated last month
• ColinQiyangLi / AdaCat
  AdaCat
  ☆49Updated 2 years ago
• nshepperd / jaxtorch
  A JAX nn library
  ☆21Updated 8 months ago
• lucidrains / mlp-gpt-jax
  A GPT, made only of MLPs, in Jax
  ☆55Updated 3 years ago
• rolandgvc / flaxvision
  A selection of neural network models ported from torchvision for JAX & Flax.
  ☆44Updated 3 years ago
• lucidrains / tranception-pytorch
  Implementation of Tranception, an attention network, paired with retrieval, that is SOTA for protein fitness prediction
  ☆31Updated 2 years ago
• lucidrains / local-attention-flax
  Local Attention - Flax module for Jax
  ☆20Updated 3 years ago
• hadyelsahar / EBM-visualizations
  notebooks of cool EBM visualizations
  ☆16Updated 3 years ago
• peerdavid / layerwise-batch-entropy
  Layerwise Batch Entropy Regularization
  ☆22Updated 2 years ago
• google-deepmind / dks
  Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…
  ☆64Updated last week
• gisilvs / AEF
  ☆33Updated last year
• j-towns / vdvae-jax
  Very deep VAEs in JAX/Flax
  ☆45Updated 3 years ago
• yk / PyTorch_CIFAR10
  Pretrained TorchVision models on CIFAR10 dataset (with weights)
  ☆24Updated 4 years ago
• ebartrum / lightning_gan_zoo
  GAN models implemented with Pytorch Lightning and Hydra configuration
  ☆34Updated 2 years ago
• fsschneider / cockpit-experiments
  Experiments for the NeurIPS 2021 paper "Cockpit: A Practical Debugging Tool for the Training of Deep Neural Networks"
  ☆13Updated 3 years ago
• stanfordmlgroup / disentanglement
  Official repository for our ICLR 2021 paper Evaluating the Disentanglement of Deep Generative Models with Manifold Topology
  ☆36Updated 3 years ago
• unixpickle / vae-textures
  Texture mapping with variational auto-encoders
  ☆40Updated 3 years ago
• nestordemeure / flaxOptimizers
  A collection of optimizers, some arcane others well known, for Flax.
  ☆29Updated 3 years ago
• shreyansh26 / ML-Optimizers-JAX
  Toy implementations of some popular ML optimizers using Python/JAX
  ☆43Updated 3 years ago
• lucidrains / ESBN-pytorch
  Usable implementation of Emerging Symbol Binding Network (ESBN), in Pytorch
  ☆23Updated 3 years ago
• Zasder3 / open_clip_juwels
  An open source implementation of CLIP.
  ☆32Updated 2 years ago
• Quansight / pytest-pytorch
  pytest plugin for a better developer experience when working with the PyTorch test suite
  ☆43Updated 2 years ago
• nathanbreitsch / torchmocks
  Test pytorch code with minimal computational overhead
  ☆25Updated last year
• ganguli-lab / degrees-of-freedom
  ☆36Updated 2 years ago
• EleutherAI / equivariance
  A framework for implementing equivariant DL
  ☆10Updated 3 years ago
• tchaton / pytorch2lightning
  ☆15Updated 3 years ago