Alternatives and similar repositories for jax-tutorial

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

• srush / jax-lda
  ☆30Updated 3 years ago
• prajjwal1 / fluence
  A deep learning library based on Pytorch focussed on low resource language research and robustness
  ☆70Updated 4 years ago
• hadyelsahar / EBM-visualizations
  notebooks of cool EBM visualizations
  ☆15Updated 4 years ago
• ethanluoyc / dhmc-jax
  Discontinuous Hamiltonian Monte Carlo in JAX
  ☆42Updated 5 years ago
• lvwerra / deep-math
  Implementation of "Analysing Mathematical Reasoning Abilities of Neural Models"
  ☆30Updated 2 years ago
• sscardapane / hydra-tutorial
  Companion code for a tutorial on using Hydra.
  ☆30Updated 4 years ago
• n2cholas / shapecheck
  Framework-agnostic library for checking array/tensor shapes at runtime.
  ☆46Updated 4 years ago
• cgraywang / transformer-on-diet
  Code repo for "Transformer on a Diet" paper
  ☆31Updated 5 years ago
• ExpectationMax / sklearn-jax-kernels
  Composable kernels for scikit-learn implemented in JAX.
  ☆45Updated 5 years ago
• shreyansh26 / ML-Optimizers-JAX
  Toy implementations of some popular ML optimizers using Python/JAX
  ☆44Updated 4 years ago
• huggingface / model_card
  ☆30Updated 4 years ago
• MathInf / toroidal
  a lightweight transformer library for PyTorch
  ☆72Updated 4 years ago
• EleutherAI / equivariance
  A framework for implementing equivariant DL
  ☆10Updated 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 4 years ago
• mblondel / projection-losses
  Python implementation of projection losses.
  ☆27Updated 6 years ago
• harvardnlp / hmm-lm
  ☆40Updated 4 years ago
• ahthie7u / cockpit
  Code for the anonymous submission "Cockpit: A Practical Debugging Tool for Training Deep Neural Networks"
  ☆31Updated 5 years ago
• probml / JSL
  Jax SSM Library
  ☆48Updated 3 years ago
• stanislavfort / adversaries_to_convnext
  Adversarial examples to the new ConvNeXt architecture
  ☆20Updated 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
• RobertTLange / code-and-blog
  Some small scale experiments for my blog posts 📝
  ☆79Updated 3 years ago
• harvardnlp / strux
  ☆18Updated 3 years ago
• widmi / widis-lstm-tools
  Tools for working with Long Short-Term Memory (LSTM) networks and sequences in Pytorch
  ☆36Updated 4 years ago
• namedtensor / notation
  ☆108Updated 2 years ago
• francescocicala / deep-learning-with-pytorch
  Deep learning lectures I am holding for the MSc on Data Science and Scientific Computing
  ☆14Updated 3 years ago
• antofuller / configaformers
  A python library for highly configurable transformers - easing model architecture search and experimentation.
  ☆49Updated 4 years ago
• teddykoker / learning-to-learn-jax
  JAX implementation of Learning to learn by gradient descent by gradient descent
  ☆28Updated 4 months ago
• google / jax-recommenders
  ☆11Updated 3 years ago
• sgugger / hf_examples
  NLP Examples using the 🤗 libraries
  ☆40Updated 4 years ago
• kachayev / dataclasses-tensor
  Easily serialize dataclasses to and from tensors (PyTorch, NumPy)
  ☆18Updated 4 years ago