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
• HomebrewML / HomebrewNLP-torch
  A case study of efficient training of large language models using commodity hardware.
  ☆69Updated 2 years ago
• abheesht17 / whisper-keras
  ☆16Updated 2 years ago
• conceptofmind / vit-flax
  Implementation of numerous Vision Transformers in Google's JAX and Flax.
  ☆22Updated 2 years ago
• sayakpaul / ViT-jax2tf
  This repository hosts code for converting the original Vision Transformer models (JAX) to TensorFlow.
  ☆33Updated 3 years ago
• LukeWood / reef-net
  ☆24Updated 2 years ago
• arampacha / reformer_fastai
  Fastai community entry to 2020 Reproducibility Challenge
  ☆17Updated 2 years ago
• sgugger / hf_examples
  NLP Examples using the 🤗 libraries
  ☆41Updated 4 years ago
• HomebrewML / Olmax
  HomebrewNLP in JAX flavour for maintable TPU-Training
  ☆50Updated last year
• sayakpaul / parameter-ensemble-differential-evolution
  Shows how to do parameter ensembling using differential evolution.
  ☆10Updated 3 years ago
• TomFrederik / grokking
  Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'
  ☆38Updated 3 years ago
• aliabd / fastgradio
  Build fast gradio demos of fastai learners
  ☆35Updated 3 years ago
• jxtngx / lightning-lab
  deep learning with pytorch lightning
  ☆1Updated 7 months ago
• rasbt / 2021-pydata-jeddah
  Materials for "Transformers from the Ground Up" at PyData Jeddah on August 5, 2021
  ☆20Updated 3 years ago
• neptune-ai / neptune-examples
  Examples of using Neptune to keep track of your experiments (maintenance only).
  ☆26Updated 3 years ago
• rasbt / cyclemoid-pytorch
  Cyclemoid implementation for PyTorch
  ☆90Updated 3 years ago
• tmabraham / fastai_tpu
  TPU support for the fastai library
  ☆13Updated 4 years ago
• sayakpaul / PAWS-TF
  Minimal implementation of PAWS (https://arxiv.org/abs/2104.13963) in TensorFlow.
  ☆45Updated 4 years ago
• lessw2020 / transformer_central
  Various transformers for FSDP research
  ☆37Updated 2 years ago
• lordtt13 / transformers-experiments
  All my experiments with the various transformers and various transformer frameworks available
  ☆14Updated 4 years ago
• elliotwaite / pytorch-hooks-tutorial
  Examples of using PyTorch hooks, as covered in my YouTube tutorial video.
  ☆33Updated last year
• lucidrains / feedback-transformer-pytorch
  Implementation of Feedback Transformer in Pytorch
  ☆107Updated 4 years ago
• twsl / whos-there
  The spiritual successor to knockknock for PyTorch Lightning, get notified when your training ends
  ☆77Updated 6 months ago
• sayakpaul / count-tokens-hf-datasets
  This project shows how to derive the total number of training tokens from a large text dataset from 🤗 datasets with Apache Beam and Data…
  ☆27Updated 2 years ago
• paperswithcode / tutorials
  Basic guidance on how to contribute to Papers with Code
  ☆23Updated 3 years ago
• stas00 / ml-ways
  ML/DL Math and Method notes
  ☆61Updated last year
• lucidrains / kalman-filtering-attention
  Implementation of the Kalman Filtering Attention proposed in "Kalman Filtering Attention for User Behavior Modeling in CTR Prediction"
  ☆58Updated last year
• Yard1 / ray-skorch
  Distributed skorch on Ray Train
  ☆57Updated 2 years ago
• lucidrains / tableformer-pytorch
  Implementation of TableFormer, Robust Transformer Modeling for Table-Text Encoding, in Pytorch
  ☆39Updated 3 years ago
• XinshaoAmosWang / DeepCriticalLearning
  Deep Critical Learning. Implementation of ProSelfLC, IMAE, DM, etc.
  ☆31Updated 2 years ago