captain-pool / hydra-example

Related projects: ⓘ

• tk-rusch / unicornn
  Official code for UnICORNN (ICML 2021)
  ☆27Updated 2 years ago
• google-deepmind / eigengame
  Open source code for EigenGame.
  ☆28Updated last year
• tk-rusch / coRNN
  Official code for Coupled Oscillatory RNN (ICLR 2021, Oral)
  ☆43Updated 3 years ago
• enajx / HyperNCA
  ☆34Updated 2 years ago
• tristandeleu / jax-meta-learning
  A collection of meta-learning algorithms in Jax
  ☆23Updated 2 years ago
• ganguli-lab / degrees-of-freedom
  ☆35Updated 2 years ago
• facebookresearch / projUNN
  Fast training of unitary deep network layers from low-rank updates
  ☆28Updated last year
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆28Updated 4 months ago
• ayulockin / LossLandscape
  Explores the ideas presented in Deep Ensembles: A Loss Landscape Perspective (https://arxiv.org/abs/1912.02757) by Stanislav Fort, Huiyi …
  ☆61Updated 4 years ago
• y0ast / pytorch-snippets
  Collection of snippets for PyTorch users
  ☆26Updated 2 years ago
• ejmichaud / grokking-squared
  ☆21Updated last year
• alexub / jax-meta-learning
  Simple, extensible implementations of some meta-learning algorithms in Jax
  ☆10Updated 3 years ago
• captain-pool / hydra-wandb-sweeper
  WandB sweeps integration with Hydra sweeper
  ☆45Updated 7 months ago
• leonard-gleyzer / connex
  Fine-grained, dynamic control of neural network topology in JAX.
  ☆21Updated last year
• cgarciae / jax_metrics
  A metrics library for the JAX ecosystem
  ☆36Updated last year
• teddykoker / learning-to-learn-jax
  JAX implementation of Learning to learn by gradient descent by gradient descent
  ☆25Updated 2 years ago
• romanpogodin / towards-bio-plausible-conv
  ☆14Updated 3 years ago
• gauravdhama / eigengame_deepmind
  A basic implementation of the paper Eigengame : PCA as a Nash Equilibrium
  ☆21Updated 3 years ago
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆36Updated 5 months ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆74Updated 7 months ago
• augustwester / gflownet
  A PyTorch implementation of a Generative Flow Network (GFlowNet) proposed by Bengio et al. (2021)
  ☆40Updated last year
• cgarciae / nanoGPT-jax
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆30Updated 9 months ago
• runame / laplace-redux
  Laplace Redux -- Effortless Bayesian Deep Learning
  ☆35Updated last year
• smonsays / equilibrium-propagation
  Fully documented Pytorch implementation of the Equilibrium Propagation algorithm.
  ☆31Updated 4 years ago
• RobertTLange / code-and-blog
  Some small scale experiments for my blog posts 📝
  ☆78Updated 2 years ago
• CUAI / Riemannian-Residual-Neural-Networks
  [NeurIPS 2023] Riemannian Residual Neural Networks (https://arxiv.org/abs/2006.10254)
  ☆14Updated 7 months ago
• thudzj / ELLA
  Code for Accelerated Linearized Laplace Approximation for Bayesian Deep Learning (ELLA, NeurIPS 22')
  ☆16Updated last year
• automl / zero-shot-automl-with-pretrained-models
  Official repository for the paper "Zero-Shot AutoML with Pretrained Models"
  ☆41Updated 8 months ago
• gengala / egnca
  Official implementation of E(n)-equivariant Graph Neural Cellular Automata
  ☆22Updated 4 months ago
• facebookresearch / w2ot
  Euclidean Wasserstein-2 optimal transportation
  ☆43Updated last year