Alternatives and similar repositories for JAX-examples

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

• stas00 / ml-ways
  ML/DL Math and Method notes
  ☆65Updated 2 years ago
• pengyan510 / torcheck
  ☆36Updated 3 years ago
• sayakpaul / BiT-jax2tf
  This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.
  ☆14Updated 4 years ago
• tensorops / TransformerX
  Flexible Python library providing building blocks (layers) for reproducible Transformers research (Tensorflow ✅, Pytorch 🔜, and Jax 🔜)
  ☆53Updated last year
• sayakpaul / ViT-jax2tf
  This repository hosts code for converting the original Vision Transformer models (JAX) to TensorFlow.
  ☆33Updated 3 years ago
• abheesht17 / whisper-keras
  ☆16Updated 2 years ago
• rasbt / cvpr2023
  ☆134Updated 2 years ago
• HomebrewML / HomebrewNLP-torch
  A case study of efficient training of large language models using commodity hardware.
  ☆68Updated 3 years ago
• rasbt / cyclemoid-pytorch
  Cyclemoid implementation for PyTorch
  ☆90Updated 3 years ago
• elliotwaite / pytorch-hooks-tutorial
  Examples of using PyTorch hooks, as covered in my YouTube tutorial video.
  ☆39Updated 2 years ago
• DarshanDeshpande / jax-models
  Unofficial JAX implementations of deep learning research papers
  ☆160Updated 3 years ago
• The-AI-Summer / learn-deep-learning
  AI Summer's complete catalog of articles
  ☆112Updated 3 years ago
• fkodom / transformer-from-scratch
  Code implementation from my blog post: https://fkodom.substack.com/p/transformers-from-scratch-in-pytorch
  ☆97Updated 2 years ago
• conceptofmind / vit-flax
  Implementation of numerous Vision Transformers in Google's JAX and Flax.
  ☆22Updated 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…
  ☆27Updated 3 years ago
• jiaweizzhao / ZerO-initialization
  ☆75Updated 3 years ago
• sayakpaul / FunMatch-Distillation
  TF2 implementation of knowledge distillation using the "function matching" hypothesis from https://arxiv.org/abs/2106.05237.
  ☆88Updated 4 years ago
• Lightning-AI / tutorials
  Collection of Pytorch lightning tutorial form as rich scripts automatically transformed to ipython notebooks.
  ☆320Updated 4 months ago
• andravin / spio
  Experimental CUDA kernel framework unifying typed dimensions, NVRTC JIT specialization, and ML‑guided tuning.
  ☆46Updated this week
• szmikler / pytorch-symbolic
  Symbolic API for model creation in PyTorch.
  ☆68Updated 9 months ago
• amirabbasasadi / RockyML
  ⛰️ RockyML - A High-Performance Scientific Computing Framework for Non-smooth Machine Learning Problems
  ☆20Updated 2 years ago
• Chris-hughes10 / pytorch-accelerated
  A lightweight library designed to accelerate the process of training PyTorch models by providing a minimal, but extensible training loop …
  ☆193Updated 6 months ago
• sayakpaul / deploy-hf-tf-vision-models
  This repository shows various ways of deploying a vision model (TensorFlow) from 🤗 Transformers.
  ☆30Updated 3 years ago
• sayakpaul / MLPMixer-jax2tf
  This repository hosts code for converting the original MLP Mixer models (JAX) to TensorFlow.
  ☆15Updated 4 years ago
• amaarora / melanoma_wandb
  ☆15Updated 3 years ago
• sayakpaul / MLP-Mixer-CIFAR10
  Implements MLP-Mixer (https://arxiv.org/abs/2105.01601) with the CIFAR-10 dataset.
  ☆59Updated 3 years ago
• johnma2006 / candle
  Deep learning library implemented from scratch in numpy. Mixtral, Mamba, LLaMA, GPT, ResNet, and other experiments.
  ☆54Updated last year
• FrancescoSaverioZuppichini / Loading-huge-PyTorch-models-with-linear-memory-consumption
  Little article showing how to load pytorch's models with linear memory consumption
  ☆34Updated 3 years ago
• dpressel / mint
  MinT: Minimal Transformer Library and Tutorials
  ☆260Updated 3 years ago
• ma7dev / deep-learning-project-template
  A project template where PyTorch Lightning, Pydantic, and more! being used for training MNIST as an example.
  ☆28Updated 3 years ago