Bennnun

posenet-pytorch

A PyTorch port of Google TensorFlow.js PoseNet (Real-time Human Pose Estimation)

terminal

The new Windows Terminal and the original Windows console host, all in the same place!

unet-seg

fastbook

The fastai book, published as Jupyter Notebooks

tnt

A lightweight library for PyTorch training tools and utilities

compress-fasttext

Tools for shrinking fastText models (in gensim format)

GraSP

Code for "Picking Winning Tickets Before Training by Preserving Gradient Flow" https://openreview.net/pdf?id=SkgsACVKPH

pytorch_influence_functions

This is a PyTorch reimplementation of Influence Functions from the ICML2017 best paper: Understanding Black-box Predictions via Influence Functions by Pang Wei Koh and Percy Liang.

1on1-questions

Mega list of 1 on 1 meeting questions compiled from a variety to sources

awesome-machine-learning-on-source-code

Cool links & research papers related to Machine Learning applied to source code (MLonCode)

DeepSpeed

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

pytorch3d

PyTorch3D is FAIR's library of reusable components for deep learning with 3D data

book_nbs

Notebooks for upcoming fastai book (draft / incomplete)

NeMo

A scalable generative AI framework built for researchers and developers working on Large Language Models, Multimodal, and Speech AI (Automatic Speech Recognition and Text-to-Speech)

inference_batched_bandits

Inference for Batched Bandits

torchnca

A PyTorch implementation of Neighbourhood Components Analysis.

python-guide

Python best practices guidebook, written for humans.

spinningup

An educational resource to help anyone learn deep reinforcement learning.

pytorch-lightning

Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes.

fastpages

An easy to use blogging platform, with enhanced support for Jupyter Notebooks.

tpu_pretrain

LM Pretraining with PyTorch/TPU

ManifoldMixup

Manifold-Mixup implementation for fastai V1

xla

Enabling PyTorch on XLA Devices (e.g. Google TPU)

MOE

A global, black box optimization engine for real world metric optimization.

facenet-pytorch

Pretrained Pytorch face detection (MTCNN) and facial recognition (InceptionResnet) models

r-cade

Retro Game Engine for Racket

fastai-v3

Starter app for fastai v3 model deployment on Render

pytorch-image-models

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

ThermalGAN

Multimodal Color-to-Thermal Image Translation

pytorch-CycleGAN-and-pix2pix

Image-to-Image Translation in PyTorch