Alternatives and similar repositories for jax-tutorial:

Users that are interested in jax-tutorial are comparing it to the libraries listed below

• michalwols / yann
  Yet Another Neural Network Library 🤔
  ☆25Updated this week
• sscardapane / hydra-tutorial
  Companion code for a tutorial on using Hydra.
  ☆29Updated 3 years ago
• shreyansh26 / ML-Optimizers-JAX
  Toy implementations of some popular ML optimizers using Python/JAX
  ☆43Updated 3 years ago
• ethanluoyc / dhmc-jax
  Discontinuous Hamiltonian Monte Carlo in JAX
  ☆41Updated 4 years ago
• alexub / jax-meta-learning
  Simple, extensible implementations of some meta-learning algorithms in Jax
  ☆9Updated 4 years ago
• arogozhnikov / adamw_bfloat16
  AdamW optimizer for bfloat16 models in pytorch 🔥.
  ☆31Updated 7 months ago
• hadyelsahar / EBM-visualizations
  notebooks of cool EBM visualizations
  ☆15Updated 3 years ago
• thiery-lab / manifold-mcmc-for-diffusions
  Code accompanying the paper 'Manifold MCMC methods for Bayesian inference in a wide class of diffusion models'
  ☆10Updated last year
• gebob19 / introduction_to_normalizing_flows
  Jupyter Notebook corresponding to 'Going with the Flow: An Introduction to Normalizing Flows'
  ☆25Updated 3 years ago
• minhtannguyen / MomentumRNN
  Code base for MomentumRNN.
  ☆18Updated 4 years ago
• EleutherAI / equivariance
  A framework for implementing equivariant DL
  ☆10Updated 3 years ago
• srush / jax-lda
  ☆30Updated 2 years ago
• lucidrains / local-attention-flax
  Local Attention - Flax module for Jax
  ☆20Updated 3 years ago
• jmtomczak / git_flow
  General Invertible Transformations for Flow-based Generative Models
  ☆17Updated 4 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
• locuslab / ase
  Analogous Safe-state Exploration (ASE) is an algorithm for provably safe and optimal exploration in MDPs with unknown, stochastic dynamic…
  ☆11Updated 3 years ago
• franrruiz / uivi
  Code for Unbiased Implicit Variational Inference (UIVI)
  ☆13Updated 6 years ago
• teddykoker / learning-to-learn-jax
  JAX implementation of Learning to learn by gradient descent by gradient descent
  ☆26Updated 3 months ago
• fastai / fastpy
  An easy way to start a python programming environment using GitHub Codespaces.
  ☆15Updated 4 years ago
• twidddj / tokmangan
  codes for TokenManipulationGAN
  ☆7Updated 4 years ago
• kyunghyuncho / backprop-kalman-filter
  ☆45Updated 5 years ago
• remilepriol / causal-adaptation-speed
  Investigate the speed of adaptation of structural causal models
  ☆16Updated 3 years ago
• ahthie7u / cockpit
  Code for the anonymous submission "Cockpit: A Practical Debugging Tool for Training Deep Neural Networks"
  ☆31Updated 4 years ago
• samiraabnar / Reflect
  Official Implementation of "Transferring Inductive Biases Through Knowledge Distillation"
  ☆14Updated 4 years ago
• ColinQiyangLi / AdaCat
  AdaCat
  ☆49Updated 2 years ago
• PAIR-code / jax-recommenders
  ☆9Updated 2 years ago
• benbo / interactive-weak-supervision
  Interactive Weak Supervision: Learning Useful Heuristics for Data Labeling
  ☆30Updated 3 years ago
• LoicGrobol / zeldarose
  Train transformer-based models.
  ☆28Updated last month
• fastai / docments
  Document parameters using comments
  ☆10Updated 3 years ago
• srush / VirtualTeaching
  DIY setup for virtual teaching on ubuntu
  ☆39Updated 4 years ago