Discover sayakpaul/MLPMixer-jax2tf Open Source project

Stars
13
Rank 1,512,455 (Top 30 %)
Language
Jupyter Notebook
License
Apache License 2.0
Created about 3 years ago
Updated over 2 years ago

sayakpaul

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

This repository hosts code for converting the original MLP Mixer models (JAX) to TensorFlow.

TF-2.0-Hacks

Contains my explorations of TensorFlow 2.x

Jupyter Notebook

383

probing-vits

Probing the representations of Vision Transformers.

Jupyter Notebook

306

diffusers-torchao

End-to-end recipes for optimizing diffusion models with torchao and diffusers (inference and FP8 training).

Python

231

ml-deployment-k8s-fastapi

This project shows how to serve an ONNX-optimized image classification model as a web service with FastAPI, Docker, and Kubernetes.

Jupyter Notebook

191

Adventures-in-TensorFlow-Lite

This repository contains notebooks that show the usage of TensorFlow Lite for quantizing deep neural networks.

Jupyter Notebook

146

maxim-tf

Implementation of MAXIM in TensorFlow.

Jupyter Notebook

130

Supervised-Contrastive-Learning-in-TensorFlow-2

Implements the ideas presented in https://arxiv.org/pdf/2004.11362v1.pdf by Khosla et al.

Jupyter Notebook

130

robustness-vit

Contains code for the paper "Vision Transformers are Robust Learners" (AAAI 2022).

Jupyter Notebook

122

MIRNet-TFLite-TRT

TensorFlow Lite models for MIRNet for low-light image enhancement.

Jupyter Notebook

113

ConvNeXt-TF

Includes PyTorch -> Keras model porting code for ConvNeXt family of models with fine-tuning and inference notebooks.

Jupyter Notebook

SimSiam-TF

Minimal implementation of SimSiam (https://arxiv.org/abs/2011.10566) in TensorFlow 2.

Jupyter Notebook

dreambooth-keras

Implementation of DreamBooth in KerasCV and TensorFlow.

Jupyter Notebook

SimCLR-in-TensorFlow-2

(Minimally) implements SimCLR (https://arxiv.org/abs/2002.05709) in TensorFlow 2.

Jupyter Notebook

single-video-curation-svd

Educational repository for applying the main video data curation techniques presented in the Stable Video Diffusion paper.

Jupyter Notebook

Adaptive-Gradient-Clipping

Minimal implementation of adaptive gradient clipping (https://arxiv.org/abs/2102.06171) in TensorFlow 2.

Jupyter Notebook

FunMatch-Distillation

TF2 implementation of knowledge distillation using the "function matching" hypothesis from https://arxiv.org/abs/2106.05237.

Jupyter Notebook

Grocery-Product-Detection

This repository builds a product detection model to recognize products from grocery shelf images.

Jupyter Notebook

stable-diffusion-keras-ft

Fine-tuning Stable Diffusion using Keras.

Python

robustness-foundation-models

This repository holds code and other relevant files for the NeurIPS 2022 tutorial: Foundational Robustness of Foundation Models.

Jupyter Notebook

Sharpness-Aware-Minimization-TensorFlow

Implements sharpness-aware minimization (https://arxiv.org/abs/2010.01412) in TensorFlow 2.

Jupyter Notebook

MLP-Mixer-CIFAR10

Implements MLP-Mixer (https://arxiv.org/abs/2105.01601) with the CIFAR-10 dataset.

Jupyter Notebook

ML-Bootcamp-Launchpad

Contains notebooks prepared for ML Bootcamp organized by Google Developers Launchpad.

Jupyter Notebook

Learnable-Image-Resizing

TF 2 implementation Learning to Resize Images for Computer Vision Tasks (https://arxiv.org/abs/2103.09950v1).

Jupyter Notebook

Knowledge-Distillation-in-Keras

Demonstrates knowledge distillation for image-based models in Keras.

Jupyter Notebook

PAWS-TF

Minimal implementation of PAWS (https://arxiv.org/abs/2104.13963) in TensorFlow.

Jupyter Notebook

swin-transformers-tf

Implementation of Swin Transformers in TensorFlow along with converted pre-trained models, code for off-the-shelf classification and fine-tuning.

Jupyter Notebook

Generalized-ODIN-TF

TensorFlow 2 implementation of the paper Generalized ODIN: Detecting Out-of-distribution Image without Learning from Out-of-distribution Data (https://arxiv.org/abs/2002.11297).

Jupyter Notebook

hf-codegen

A repository of Python scripts to scrape code contents of the public repositories of `huggingface`.

Python

A-B-testing-with-Machine-Learning

Implemented an A/B Testing solution with the help of machine learning

Jupyter Notebook

deit-tf

Includes PyTorch -> Keras model porting code for DeiT models with fine-tuning and inference notebooks.

Jupyter Notebook

Barlow-Twins-TF

TensorFlow implementation of Barlow Twins (https://arxiv.org/abs/2103.03230).

Jupyter Notebook

Vehicle-Number-Plate-Detection

This project demonstrates the use of TensorFlow Object Detection API to automatically number plates (Indian) from vehicles

Jupyter Notebook

Benchmarking-and-MLI-experiments-on-the-Adult-dataset

Contains benchmarking and interpretability experiments on the Adult dataset using several libraries

Jupyter Notebook

Dual-Deployments-on-Vertex-AI

Project demonstrating dual model deployment scenarios using Vertex AI (GCP).

Jupyter Notebook

ViT-jax2tf

This repository hosts code for converting the original Vision Transformer models (JAX) to TensorFlow.

Jupyter Notebook

Breast-Cancer-Detection-using-Deep-Learning

Experiments to show the usage of deep learning to detect breast cancer from breast histopathology images

Jupyter Notebook

Spatial-Transformer-Networks-with-Keras

This repository provides a Colab Notebook that shows how to use Spatial Transformer Networks inside CNNs in Keras.

Jupyter Notebook

deploy-hf-tf-vision-models

This repository shows various ways of deploying a vision model (TensorFlow) from 🤗 Transformers.

Jupyter Notebook

E2E-Object-Detection-in-TFLite

This repository shows how to train a custom detection model with the TFOD API, optimize it with TFLite, and perform inference with the optimized model.

Jupyter Notebook

A-Barebones-Image-Retrieval-System

This project presents a simple framework to retrieve images similar to a query image.

Jupyter Notebook

Training-BatchNorm-and-Only-BatchNorm

Experiments with the ideas presented in https://arxiv.org/abs/2003.00152 by Frankle et al.

Jupyter Notebook

ml-bootcamp-india-2022

Contains materials from the facilitation sessions conducted for the ML Bootcamp India (2022) organized by Google DevRel team.

Jupyter Notebook

BERT-for-Mobile

Compares the DistilBERT and MobileBERT architectures for mobile deployments.

Jupyter Notebook

Blood-Cell-Detection-using-TFOD-API

This project demonstrates the use of TensorFlow Object Detection API (along with GCP ML Engine) to automatically detect Red Blood Cells (RBCs), White Blood Cells (WBCs), and Platelets in each image taken via microscopic image readings

Jupyter Notebook

CI-CD-for-Model-Training

This repository holds files and scripts for incorporating simple CI/CD practices for model training in ML.

Jupyter Notebook

big_vision_experiments

Contains my experiments with the `big_vision` repo to train ViTs on ImageNet-1k.

Jupyter Notebook

Denoised-Smoothing-TF

Minimal implementation of Denoised Smoothing (https://arxiv.org/abs/2003.01908) in TensorFlow.

Jupyter Notebook

portfolio

Site built from fastpages: https://fastpages.fast.ai/. Deployed here 👉

Jupyter Notebook

Global-Wheat-Detection

Showcases the use of deep learning to detect wheat heads from crops. The project is based on: https://www.kaggle.com/c/global-wheat-detection.

Jupyter Notebook

Distributed-Training-in-TensorFlow-2-with-AI-Platform

Contains code to demonstrate distributed training in TensorFlow 2 with AI Platform and custom Docker contains.

Python

instructpix2pix-sdxl

Training InstructPi2Pix with SDXL.

Python

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 Dataflow.

Python

FloydHub-Anomaly-Detection-Blog

Contains the thorough experiments made for a FloydHub article on Anomaly Detection

Jupyter Notebook

Phishing-Websites-Detection

Experiments to detect phishing websites using neural networks

Jupyter Notebook

Malaria-Detection-with-Deep-Learning

Deep learning based solution to automatically analyze medical images for malaria testing

Jupyter Notebook

BiT-jax2tf

This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.

Jupyter Notebook

Parkinson-s-Disease-Classifier

Deep learning experiments to design a model to predict Parkinson's diseases with the images of Spiral/Wave test

Jupyter Notebook

instruct-pix2pix-dataset

This repository provides utilities to a minimal dataset for InstructPix2Pix like training for Diffusion models.

Python

Generating-categories-from-arXiv-paper-titles

This project takes the arXiv dataset and builds an automatic tag classifier from the arXiv article/paper titles

Jupyter Notebook

Handwriting-Recognizer-in-Keras

This project shows how to build a simple handwriting recognizer in Keras with the IAM dataset.

Jupyter Notebook

Mixed-Precision-Training-in-tf.keras-2.0

This repository contains notebooks showing how to perform mixed precision training in tf.keras 2.0

Jupyter Notebook

NALU

Neural Arithmetic Logic Units by Trask et al.

Jupyter Notebook

Adversarial-Examples-in-Deep-Learning

Shows how to create basic image adversaries, and train adversarially robust image classifiers (to some extent).

Jupyter Notebook

FloydHub-Q-Learning-Blog

Contains the Jupyter Notebook made for a FloydHub article on Q-Learning

Jupyter Notebook

EvoNorms-in-TensorFlow-2

Implements EvoNorms B0 and S0 as proposed in Evolving Normalization-Activation Layers.

Jupyter Notebook

Revisiting-Pooling-in-CNNs

Implements RNNPool and SoftPool for CNNs.

Jupyter Notebook

parameter-ensemble-differential-evolution

Shows how to do parameter ensembling using differential evolution.

Jupyter Notebook

Action-Recognition-in-TensorFlow

Contains additional materials for two keras.io blog posts.

Jupyter Notebook

AdaMatch-TF

Includes additional materials for the following keras.io blog post.

Jupyter Notebook

keras-convnext-conversion

ConvNeXt conversion code for PT to TF along with evaluation code on ImageNet-1k val.

Python

Emotion-Detection-using-Deep-Learning

This project demonstrates the use of Deep Learning to detect emotion (sad, angry, happy etc) from the images of faces.

Jupyter Notebook

Transfer-Learning-with-CIFAR10

Leveraging Transfer Learning on the classic CIFAR-10 dataset by using the weights from a pre-trained VGG-16 model.

Jupyter Notebook

TalksGiven

Contains the deck of my talks given at different developer meet-ups and conferences.

cait-tf

Implementation of CaiT models in TensorFlow and ImageNet-1k checkpoints. Includes code for inference and fine-tuning.

Jupyter Notebook

GCP-ML-API-Demos

Contains Colab Notebooks show cool use-cases of different GCP ML APIs.

Jupyter Notebook

DockerML

Contains my explorations of using Docker to automate ML workflows.

Python

Multimodal-Entailment-Baseline

This repository shows how to implement a basic model for multimodal entailment.

Jupyter Notebook

vision-transformers-tf

A non-exhaustive collection of vision transformer models implemented in TensorFlow.

xla-benchmark-sd

Provides code to serialize the different models involved in Stable Diffusion as SavedModels and to compile them with XLA.

Python

tf.keras-Distributed-Training

Shows how to use MirroredStrategy to distribute training workloads when using the regular fit and compile paradigm in tf.keras.

Jupyter Notebook

Analytics-Vidhya-Game-of-Deep-Learning-Hackathon

Contains my experiments for the Game of Deep Learning Hackathon conducted by Analytics Vidhya

Jupyter Notebook

model-soups-text-classification

Shows an implementation of model soups (https://arxiv.org/abs/2203.05482) for text classification models.

Jupyter Notebook

keras-xla-benchmarks

Presents comprehensive benchmarks of XLA-compatible pre-trained models in Keras.

Python

Manning-Phishing-Websites-Detection

Starter repository for Manning LP: Use Machine Learning to Detect Phishing Websites

Data-Science-Tweets-Analysis

Byte sized analysis of Data Science Tweets

Jupyter Notebook

Data-Pipelines-with-TensorFlow-Data-Services-Exercises

This repository contains the exercise notebooks for the Data Pipelines with TensorFlow Data Services (Coursera) course.

Jupyter Notebook

Applied-Data-Science-w-Python-Specialization

Contains my assignments, guiding notebooks (provided as the course materials) and the datasets.

Jupyter Notebook

Reproducibility-in-tf.keras-with-wandb

Contains code for my model reproducibility experiments. Report attached.

Jupyter Notebook

GDE-Collaboration

Contains the modelling and deployment scripts done for a collaboration project (Dataset: FashionMNIST)

Jupyter Notebook

simple-image-recaptioning

Recaption large (Web)Datasets with vllm and save the artifacts.

Python

Consistency-Training-with-Supervision

Contains experimentation notebooks for my Keras Example "Consistency Training with Supervision".

Jupyter Notebook

mlplanner

Contains data, notebooks and other files of FloydHub's mini-series on machine learning project structuring, model debugging, various tips and tricks and more

Jupyter Notebook

tSNE-on-NSL_KDD

Jupyter Notebook

Generating-Word-Cloud-from-DataCamp-Project-Descriptions

Generating a word cloud from the descriptions of the live [DataCamp Projects](https://datacamp.com/projects/).

Jupyter Notebook

Analysis-of-college-database-of-2017-passouts

Contains my analysis of a database containing information about the students of an engineering college.

Jupyter Notebook

Age-Detection-of-Indian-Actors

The repository contains my experiments for a Hackathon problem by Analytics Vidhya

Jupyter Notebook

depth_estimation_trainer

Scripts to fine-tune a depth estimation model.

Python

Restaurant

It contains everything that should be there in a standard restaurant website such as Location finder, Chaining, Bill Estimator etc. Built using JSP, jQuery, JavaScript, HTML-CSS and for databases Oracle 10g XE was used.

HTML

Reporting-in-SQL-DataCamp-Course-Exercises

Contains my solutions to the exercises of the course "Reporting in SQL"

100

DataCamp-blogs

Jupyter notebooks of my DataCamp blogs

Jupyter Notebook

sayakpaul/MLPMixer-jax2tf

sayakpaul

Reviews

Repository Details

More Repositories