Alternatives and similar repositories for JAX-examples

Users that are interested in JAX-examples are comparing it to the libraries listed below

• rasbt / cyclemoid-pytorch
  Cyclemoid implementation for PyTorch
  ☆89Updated 3 years ago
• sayakpaul / ViT-jax2tf
  This repository hosts code for converting the original Vision Transformer models (JAX) to TensorFlow.
  ☆33Updated 3 years ago
• HomebrewML / HomebrewNLP-torch
  A case study of efficient training of large language models using commodity hardware.
  ☆69Updated 2 years ago
• arampacha / reformer_fastai
  Fastai community entry to 2020 Reproducibility Challenge
  ☆18Updated 2 years ago
• sayakpaul / BiT-jax2tf
  This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.
  ☆14Updated 3 years ago
• soumik12345 / multi-label-text-classification
  A multi-label text classifier to predict the subject areas of arXiv papers from their abstract bodies.
  ☆17Updated 3 years ago
• abheesht17 / whisper-keras
  ☆16Updated 2 years ago
• LukeWood / reef-net
  ☆24Updated 2 years ago
• HomebrewML / Olmax
  HomebrewNLP in JAX flavour for maintable TPU-Training
  ☆50Updated last year
• lucidrains / feedback-transformer-pytorch
  Implementation of Feedback Transformer in Pytorch
  ☆107Updated 4 years ago
• jxtngx / lightning-lab
  deep learning with pytorch lightning
  ☆1Updated 7 months ago
• TomFrederik / grokking
  Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'
  ☆38Updated 3 years ago
• sgugger / hf_examples
  NLP Examples using the 🤗 libraries
  ☆41Updated 4 years ago
• cceyda / lit-NER
  TorchServe+Streamlit for easily serving your HuggingFace NER models
  ☆33Updated 2 years ago
• The-AI-Summer / RNN_tutorial
  Recurrent neural networks: building a custom LSTM/GRU cell in PyTorch
  ☆28Updated 4 years ago
• wilson1yan / cs294-158-ssl
  ☆14Updated 2 years ago
• muellerzr / fastdebug
  In-place debugger for the fastai library and pytorch
  ☆28Updated 3 years ago
• Yard1 / ray-skorch
  Distributed skorch on Ray Train
  ☆57Updated 2 years ago
• lucidrains / memory-editable-transformer
  My explorations into editing the knowledge and memories of an attention network
  ☆35Updated 2 years ago
• yiyixuxu / n-grammer-flax
  Implementation of N-Grammer in Flax
  ☆17Updated 2 years ago
• lucidrains / panoptic-transformer
  Another attempt at a long-context / efficient transformer by me
  ☆38Updated 3 years ago
• hunar4321 / RLS-neural-net
  Recursive Leasting Squares (RLS) with Neural Network for fast learning
  ☆52Updated last year
• neptune-ai / neptune-examples
  Examples of using Neptune to keep track of your experiments (maintenance only).
  ☆26Updated 3 years ago
• lucidrains / metaformer-gpt
  Implementation of Metaformer, but in an autoregressive manner
  ☆25Updated 2 years ago
• lucidrains / token-shift-gpt
  Implementation of Token Shift GPT - An autoregressive model that solely relies on shifting the sequence space for mixing
  ☆50Updated 3 years ago
• stas00 / ml-ways
  ML/DL Math and Method notes
  ☆61Updated last year
• ptrblck / tutorials
  ☆9Updated 2 years ago
• paperswithcode / tutorials
  Basic guidance on how to contribute to Papers with Code
  ☆23Updated 3 years ago
• lucidrains / einops-exts
  Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️
  ☆53Updated 2 years ago
• YeonwooSung / GLOM
  PyTorch implementation of GLOM
  ☆22Updated 3 years ago