Alternatives and similar repositories for JAX-examples

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

• sayakpaul / BiT-jax2tf
  This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.
  ☆14Updated 3 years ago
• sayakpaul / ViT-jax2tf
  This repository hosts code for converting the original Vision Transformer models (JAX) to TensorFlow.
  ☆33Updated 3 years ago
• sayakpaul / MLP-Mixer-CIFAR10
  Implements MLP-Mixer (https://arxiv.org/abs/2105.01601) with the CIFAR-10 dataset.
  ☆57Updated 3 years ago
• stas00 / ml-ways
  ML/DL Math and Method notes
  ☆63Updated last year
• tensorops / TransformerX
  Flexible Python library providing building blocks (layers) for reproducible Transformers research (Tensorflow ✅, Pytorch 🔜, and Jax 🔜)
  ☆53Updated last year
• rasbt / cyclemoid-pytorch
  Cyclemoid implementation for PyTorch
  ☆90Updated 3 years ago
• HomebrewML / HomebrewNLP-torch
  A case study of efficient training of large language models using commodity hardware.
  ☆68Updated 3 years ago
• lucidrains / mlp-gpt-jax
  A GPT, made only of MLPs, in Jax
  ☆58Updated 4 years ago
• sayakpaul / PAWS-TF
  Minimal implementation of PAWS (https://arxiv.org/abs/2104.13963) in TensorFlow.
  ☆45Updated 4 years ago
• amaarora / melanoma_wandb
  ☆15Updated 3 years ago
• HomebrewML / Olmax
  HomebrewNLP in JAX flavour for maintable TPU-Training
  ☆50Updated last year
• elliotwaite / pytorch-hooks-tutorial
  Examples of using PyTorch hooks, as covered in my YouTube tutorial video.
  ☆37Updated last year
• lessw2020 / transformer_central
  Various transformers for FSDP research
  ☆38Updated 2 years ago
• butchland / fastai_xla_extensions
  A Python package to allow fastai to run on TPUs using Pytorch-XLA
  ☆36Updated 2 years ago
• The-AI-Summer / learn-deep-learning
  AI Summer's complete catalog of articles
  ☆111Updated 3 years ago
• aliabd / fastgradio
  Build fast gradio demos of fastai learners
  ☆35Updated 3 years ago
• XinshaoAmosWang / DeepCriticalLearning
  Deep Critical Learning. Implementation of ProSelfLC, IMAE, DM, etc.
  ☆31Updated 2 years ago
• jiaweizzhao / ZerO-initialization
  ☆75Updated 2 years ago
• sayakpaul / MLPMixer-jax2tf
  This repository hosts code for converting the original MLP Mixer models (JAX) to TensorFlow.
  ☆15Updated 3 years ago
• pengyan510 / torcheck
  ☆36Updated 3 years ago
• conceptofmind / vit-flax
  Implementation of numerous Vision Transformers in Google's JAX and Flax.
  ☆22Updated 3 years ago
• arampacha / reformer_fastai
  Fastai community entry to 2020 Reproducibility Challenge
  ☆17Updated 2 years ago
• sayakpaul / big_vision_experiments
  Contains my experiments with the `big_vision` repo to train ViTs on ImageNet-1k.
  ☆22Updated 2 years ago
• wandb / Groundbreaking-Papers
  ML Research paper summaries, annotated papers and implementation walkthroughs
  ☆113Updated 3 years ago
• DarshanDeshpande / jax-models
  Unofficial JAX implementations of deep learning research papers
  ☆156Updated 3 years ago
• albanie / algorithms-and-data-structures
  Video descriptions and minimalist Python implementations of algorithms and data structures.
  ☆69Updated last year
• nreHieW / loss
  Visualising Losses in Deep Neural Networks
  ☆16Updated last year
• abheesht17 / whisper-keras
  ☆16Updated 2 years ago
• LukeWood / reef-net
  ☆24Updated 3 years ago
• AnswerDotAI / minai
  A miniture AI training framework for PyTorch
  ☆42Updated 7 months ago