The-AI-Summer / JAX-examples

Related projects ⓘ

Alternatives and complementary repositories for JAX-examples

• sayakpaul / ViT-jax2tf
  This repository hosts code for converting the original Vision Transformer models (JAX) to TensorFlow.
  ☆34Updated 2 years ago
• mlflow / mlflow-torchserve
  Plugin for deploying MLflow models to TorchServe
  ☆106Updated last year
• rasbt / cyclemoid-pytorch
  Cyclemoid implementation for PyTorch
  ☆87Updated 2 years ago
• rasbt / pytorch-fabric-demo
  ☆24Updated last year
• lucidrains / feedback-transformer-pytorch
  Implementation of Feedback Transformer in Pytorch
  ☆104Updated 3 years ago
• stas00 / ml-ways
  ML/DL Math and Method notes
  ☆57Updated 11 months ago
• jxtngx / lightning-lab
  deep learning with pytorch lightning
  ☆0Updated 3 weeks ago
• LukeWood / reef-net
  ☆24Updated 2 years ago
• TomFrederik / grokking
  Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'
  ☆38Updated 2 years ago
• HomebrewNLP / HomebrewNLP
  A case study of efficient training of large language models using commodity hardware.
  ☆68Updated 2 years ago
• sayakpaul / BiT-jax2tf
  This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.
  ☆14Updated 2 years ago
• SkafteNicki / pl_crossvalidate
  ☆60Updated 11 months ago
• conceptofmind / vit-flax
  Implementation of numerous Vision Transformers in Google's JAX and Flax.
  ☆20Updated 2 years ago
• sgugger / hf_examples
  NLP Examples using the 🤗 libraries
  ☆42Updated 3 years ago
• borisdayma / lightning-kitti
  Semantic Segmentation with Pytorch-Lightning
  ☆63Updated 3 years 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…
  ☆24Updated 2 years ago
• HomebrewNLP / Olmax
  HomebrewNLP in JAX flavour for maintable TPU-Training
  ☆46Updated 10 months ago
• abheesht17 / whisper-keras
  ☆16Updated last year
• rasbt / 2021-pydata-jeddah
  Materials for "Transformers from the Ground Up" at PyData Jeddah on August 5, 2021
  ☆18Updated 3 years ago
• sayakpaul / PAWS-TF
  Minimal implementation of PAWS (https://arxiv.org/abs/2104.13963) in TensorFlow.
  ☆45Updated 3 years ago
• arampacha / reformer_fastai
  Fastai community entry to 2020 Reproducibility Challenge
  ☆17Updated 2 years ago
• paperswithcode / tutorials
  Basic guidance on how to contribute to Papers with Code
  ☆20Updated 2 years ago
• PrototypeML / PrototypeML
  ☆30Updated 4 years ago
• huggingface / hf_benchmarks
  A starter kit for evaluating benchmarks on the 🤗 Hub
  ☆13Updated 10 months ago
• r0mainK / outperformer
  Code for scaling Transformers
  ☆26Updated 3 years ago
• thevasudevgupta / speech-jax
  Speech in Flax/JAX
  ☆15Updated 2 years ago
• The-AI-Summer / RNN_tutorial
  Recurrent neural networks: building a custom LSTM/GRU cell in PyTorch
  ☆28Updated 4 years ago
• Yard1 / ray-skorch
  Distributed skorch on Ray Train
  ☆57Updated 2 years ago
• The-AI-Summer / learn-deep-learning
  AI Summer's complete catalog of articles
  ☆107Updated 2 years ago