Alternatives and similar repositories for jax-tutorial

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

• prajjwal1 / fluence
  A deep learning library based on Pytorch focussed on low resource language research and robustness
  ☆70Updated 3 years ago
• srush / jax-lda
  ☆30Updated 3 years ago
• michalwols / yann
  Yet Another Neural Network Library 🤔
  ☆27Updated 4 months ago
• cgraywang / transformer-on-diet
  Code repo for "Transformer on a Diet" paper
  ☆31Updated 5 years ago
• rasbt / 2021-pydata-jeddah
  Materials for "Transformers from the Ground Up" at PyData Jeddah on August 5, 2021
  ☆20Updated 4 years ago
• sscardapane / hydra-tutorial
  Companion code for a tutorial on using Hydra.
  ☆30Updated 4 years ago
• shreyansh26 / ML-Optimizers-JAX
  Toy implementations of some popular ML optimizers using Python/JAX
  ☆44Updated 4 years ago
• yngtdd / hyperspace
  Distributed Bayesian Optimization
  ☆23Updated 5 years ago
• MathInf / toroidal
  a lightweight transformer library for PyTorch
  ☆72Updated 3 years ago
• sgugger / hf_examples
  NLP Examples using the 🤗 libraries
  ☆41Updated 4 years ago
• ExpectationMax / sklearn-jax-kernels
  Composable kernels for scikit-learn implemented in JAX.
  ☆44Updated 4 years 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 5 years ago
• lvwerra / deep-math
  Implementation of "Analysing Mathematical Reasoning Abilities of Neural Models"
  ☆30Updated 2 years ago
• lucidrains / local-attention-flax
  Local Attention - Flax module for Jax
  ☆22Updated 4 years ago
• Popgun-Labs / PopGen
  A generative modelling toolkit for PyTorch.
  ☆70Updated 3 years ago
• ColinQiyangLi / AdaCat
  AdaCat
  ☆49Updated 3 years ago
• arogozhnikov / adamw_bfloat16
  AdamW optimizer for bfloat16 models in pytorch 🔥.
  ☆35Updated last year
• n2cholas / shapecheck
  Framework-agnostic library for checking array/tensor shapes at runtime.
  ☆46Updated 4 years ago
• francescocicala / deep-learning-with-pytorch
  Deep learning lectures I am holding for the MSc on Data Science and Scientific Computing
  ☆13Updated 3 years ago
• edwardjhu / improved_wasserstein
  Code for our ICLR Trustworthy ML 2020 workshop paper "Improved Image Wasserstein Attacks and Defenses"
  ☆14Updated 5 years ago
• harvardnlp / hmm-lm
  ☆41Updated 4 years ago
• huggingface / model_card
  ☆30Updated 3 years ago
• hadyelsahar / EBM-visualizations
  notebooks of cool EBM visualizations
  ☆15Updated 4 years ago
• google-research / precondition
  ☆31Updated last month
• inkawhich / pt-distributed-tutorial
  ☆17Updated 6 years ago
• ethanluoyc / dhmc-jax
  Discontinuous Hamiltonian Monte Carlo in JAX
  ☆42Updated 5 years ago
• probml / JSL
  Jax SSM Library
  ☆49Updated 2 years ago
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆39Updated last year
• lucidrains / mogrifier
  Usable implementation of Mogrifier, a circuit for enhancing LSTMs and potentially other networks, from Deepmind
  ☆19Updated last year