Alternatives and similar repositories for JAX-examples

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

• stas00 / ml-ways
  ML/DL Math and Method notes
  ☆66Updated 2 years ago
• elliotwaite / pytorch-hooks-tutorial
  Examples of using PyTorch hooks, as covered in my YouTube tutorial video.
  ☆39Updated 2 years ago
• rasbt / cyclemoid-pytorch
  Cyclemoid implementation for PyTorch
  ☆90Updated 3 years ago
• abheesht17 / whisper-keras
  ☆16Updated 2 years ago
• The-AI-Summer / learn-deep-learning
  AI Summer's complete catalog of articles
  ☆112Updated 4 years ago
• sayakpaul / PAWS-TF
  Minimal implementation of PAWS (https://arxiv.org/abs/2104.13963) in TensorFlow.
  ☆45Updated 4 years ago
• sayakpaul / MLP-Mixer-CIFAR10
  Implements MLP-Mixer (https://arxiv.org/abs/2105.01601) with the CIFAR-10 dataset.
  ☆59Updated 3 years ago
• sayakpaul / ViT-jax2tf
  This repository hosts code for converting the original Vision Transformer models (JAX) to TensorFlow.
  ☆33Updated 3 years ago
• warner-benjamin / fastxtend
  Train fastai models faster (and other useful tools)
  ☆72Updated 8 months ago
• HomebrewML / HomebrewNLP-torch
  A case study of efficient training of large language models using commodity hardware.
  ☆68Updated 3 years ago
• aliabd / fastgradio
  Build fast gradio demos of fastai learners
  ☆35Updated 4 years ago
• butchland / fastai_xla_extensions
  A Python package to allow fastai to run on TPUs using Pytorch-XLA
  ☆36Updated 2 years ago
• LukeWood / reef-net
  ☆24Updated 3 years ago
• sayakpaul / MLPMixer-jax2tf
  This repository hosts code for converting the original MLP Mixer models (JAX) to TensorFlow.
  ☆15Updated 4 years ago
• pengyan510 / torcheck
  ☆36Updated 3 years ago
• reddragon / efficient-dl-survey-paper
  Efficient Deep Learning Survey Paper
  ☆34Updated 2 years ago
• HomebrewML / Olmax
  HomebrewNLP in JAX flavour for maintable TPU-Training
  ☆51Updated 2 years ago
• fkodom / transformer-from-scratch
  Code implementation from my blog post: https://fkodom.substack.com/p/transformers-from-scratch-in-pytorch
  ☆97Updated 2 years ago
• sayakpaul / deploy-hf-tf-vision-models
  This repository shows various ways of deploying a vision model (TensorFlow) from 🤗 Transformers.
  ☆30Updated 3 years ago
• sayakpaul / FunMatch-Distillation
  TF2 implementation of knowledge distillation using the "function matching" hypothesis from https://arxiv.org/abs/2106.05237.
  ☆88Updated 4 years ago
• wandb / Groundbreaking-Papers
  ML Research paper summaries, annotated papers and implementation walkthroughs
  ☆114Updated 3 years ago
• NielsRogge / Vision-Transformer-papers
  This repository contains an overview of important follow-up works based on the original Vision Transformer (ViT) by Google.
  ☆192Updated 4 years ago
• rasbt / cvpr2023
  ☆132Updated 2 years ago
• rasbt / faster-pytorch-blog
  Outlining techniques for improving the training performance of your PyTorch model without compromising its accuracy
  ☆128Updated 2 years ago
• DarshanDeshpande / jax-models
  Unofficial JAX implementations of deep learning research papers
  ☆161Updated 3 years ago
• FasterAI-Labs / fasterai
  FasterAI: Prune and Distill your models with FastAI and PyTorch
  ☆252Updated this week
• Lightning-AI / tutorials
  Collection of Pytorch lightning tutorial form as rich scripts automatically transformed to ipython notebooks.
  ☆319Updated 6 months ago
• Jingjing-NLP / KNAS
  Codes for paper "KNAS: Green Neural Architecture Search"
  ☆93Updated 4 years ago
• tensorops / TransformerX
  Flexible Python library providing building blocks (layers) for reproducible Transformers research (Tensorflow ✅, Pytorch 🔜, and Jax 🔜)
  ☆53Updated 2 years ago
• The-AI-Summer / RNN_tutorial
  Recurrent neural networks: building a custom LSTM/GRU cell in PyTorch
  ☆28Updated 5 years ago