Alternatives and similar repositories for JAX-examples

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

• abheesht17 / whisper-keras
  ☆16Updated 2 years ago
• 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
• LukeWood / reef-net
  ☆24Updated 2 years ago
• sayakpaul / PAWS-TF
  Minimal implementation of PAWS (https://arxiv.org/abs/2104.13963) in TensorFlow.
  ☆45Updated 4 years ago
• vdouet / Discriminative-learning-rates-PyTorch
  Adaptation of discriminative learning rates from the Fastai library for standard PyTorch.
  ☆18Updated 5 years ago
• rasbt / cyclemoid-pytorch
  Cyclemoid implementation for PyTorch
  ☆90Updated 3 years ago
• sayakpaul / cait-tf
  Implementation of CaiT models in TensorFlow and ImageNet-1k checkpoints. Includes code for inference and fine-tuning.
  ☆12Updated 2 years ago
• pengyan510 / torcheck
  ☆37Updated 2 years ago
• HomebrewML / HomebrewNLP-torch
  A case study of efficient training of large language models using commodity hardware.
  ☆68Updated 2 years ago
• stas00 / ml-ways
  ML/DL Math and Method notes
  ☆61Updated last year
• arampacha / reformer_fastai
  Fastai community entry to 2020 Reproducibility Challenge
  ☆17Updated 2 years ago
• jxtngx / lightning-lab
  deep learning with pytorch lightning
  ☆1Updated 8 months ago
• mrinath123 / timmFasterRcnn
  Adds timm pretrained backbone to pytorch's FasterRcnn model
  ☆12Updated last year
• sayakpaul / MLP-Mixer-CIFAR10
  Implements MLP-Mixer (https://arxiv.org/abs/2105.01601) with the CIFAR-10 dataset.
  ☆56Updated 3 years ago
• nateraw / huggingface-datasets-converter
  Scripts to convert datasets from various sources to Hugging Face Datasets.
  ☆57Updated 2 years ago
• sayakpaul / keras-xla-benchmarks
  Presents comprehensive benchmarks of XLA-compatible pre-trained models in Keras.
  ☆37Updated last year
• sayakpaul / parameter-ensemble-differential-evolution
  Shows how to do parameter ensembling using differential evolution.
  ☆10Updated 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
• elliotwaite / pytorch-hooks-tutorial
  Examples of using PyTorch hooks, as covered in my YouTube tutorial video.
  ☆34Updated last year
• Rishit-dagli / GLOM-TensorFlow
  An attempt at the implementation of GLOM, Geoffrey Hinton's paper for emergent part-whole hierarchies from data
  ☆37Updated 4 years ago
• sayakpaul / MLPMixer-jax2tf
  This repository hosts code for converting the original MLP Mixer models (JAX) to TensorFlow.
  ☆15Updated 3 years ago
• seanswyi / transformer-implementation
  Personal implementation of the Transformer paper.
  ☆22Updated last year
• HomebrewML / Olmax
  HomebrewNLP in JAX flavour for maintable TPU-Training
  ☆50Updated last year
• sjmikler / pytorch-symbolic
  Symbolic API for model creation in PyTorch.
  ☆66Updated 4 months ago
• wandb / Groundbreaking-Papers
  ML Research paper summaries, annotated papers and implementation walkthroughs
  ☆113Updated 3 years ago
• sgugger / hf_examples
  NLP Examples using the 🤗 libraries
  ☆41Updated 4 years ago
• lucidrains / mlp-gpt-jax
  A GPT, made only of MLPs, in Jax
  ☆58Updated 4 years ago
• soumik12345 / wandb-addons
  Weights & Biases Addons is a repository consisting of additional unitilities and community contributions for supercharging your Weights &…
  ☆23Updated last year
• lucidrains / feedback-transformer-pytorch
  Implementation of Feedback Transformer in Pytorch
  ☆107Updated 4 years ago