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
• LukeWood / reef-net
  ☆24Updated 2 years ago
• abheesht17 / whisper-keras
  ☆16Updated 2 years ago
• sayakpaul / parameter-ensemble-differential-evolution
  Shows how to do parameter ensembling using differential evolution.
  ☆10Updated 3 years ago
• HomebrewML / HomebrewNLP-torch
  A case study of efficient training of large language models using commodity hardware.
  ☆69Updated 2 years ago
• HomebrewML / Olmax
  HomebrewNLP in JAX flavour for maintable TPU-Training
  ☆50Updated last year
• rasbt / cyclemoid-pytorch
  Cyclemoid implementation for PyTorch
  ☆90Updated 3 years ago
• lucidrains / feedback-transformer-pytorch
  Implementation of Feedback Transformer in Pytorch
  ☆105Updated 4 years ago
• TomFrederik / grokking
  Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'
  ☆38Updated 3 years ago
• thevasudevgupta / speech-jax
  Speech in Flax/JAX
  ☆15Updated 2 years ago
• matthias-wright / jax-fid
  FID computation in Jax/Flax.
  ☆27Updated 10 months ago
• vdouet / Discriminative-learning-rates-PyTorch
  Adaptation of discriminative learning rates from the Fastai library for standard PyTorch.
  ☆18Updated 5 years ago
• sayakpaul / PAWS-TF
  Minimal implementation of PAWS (https://arxiv.org/abs/2104.13963) in TensorFlow.
  ☆45Updated 3 years ago
• sayakpaul / deploy-hf-tf-vision-models
  This repository shows various ways of deploying a vision model (TensorFlow) from 🤗 Transformers.
  ☆30Updated 2 years ago
• lucidrains / panoptic-transformer
  Another attempt at a long-context / efficient transformer by me
  ☆38Updated 3 years ago
• shawwn / ml-notes
  ☆39Updated 2 years ago
• lordtt13 / transformers-experiments
  All my experiments with the various transformers and various transformer frameworks available
  ☆14Updated 4 years ago
• conceptofmind / vit-flax
  Implementation of numerous Vision Transformers in Google's JAX and Flax.
  ☆22Updated 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
• sayakpaul / ViT-jax2tf
  This repository hosts code for converting the original Vision Transformer models (JAX) to TensorFlow.
  ☆33Updated 3 years ago
• lucidrains / mlp-gpt-jax
  A GPT, made only of MLPs, in Jax
  ☆58Updated 3 years ago
• muellerzr / fastdebug
  In-place debugger for the fastai library and pytorch
  ☆28Updated 3 years ago
• sayakpaul / cait-tf
  Implementation of CaiT models in TensorFlow and ImageNet-1k checkpoints. Includes code for inference and fine-tuning.
  ☆12Updated last year
• lucidrains / metaformer-gpt
  Implementation of Metaformer, but in an autoregressive manner
  ☆24Updated 2 years ago
• jiaweizzhao / ZerO-initialization
  ☆74Updated 2 years ago
• nateraw / spaces-template
  A 🔥 cookiecutter template for building Hugging Face Spaces
  ☆11Updated 3 years ago
• sayakpaul / MLPMixer-jax2tf
  This repository hosts code for converting the original MLP Mixer models (JAX) to TensorFlow.
  ☆15Updated 3 years ago
• ariG23498 / timm-wrapper-examples
  Notebooks to demonstrate TimmWrapper
  ☆16Updated 4 months ago
• sayakpaul / Distributed-Training-in-TensorFlow-2-with-AI-Platform
  Contains code to demonstrate distributed training in TensorFlow 2 with AI Platform and custom Docker contains.
  ☆20Updated 4 years ago
• arampacha / reformer_fastai
  Fastai community entry to 2020 Reproducibility Challenge
  ☆18Updated 2 years ago