Schedule-Free Optimization in PyTorch

リアルタイムボイスチェンジャー Realtime Voice Changer

Neighborhood Attention Extension. Bringing attention to a neighborhood near you!

Fast & Simple repository for pre-training and fine-tuning T5-style models

webui extension of ML-Danbooru

Memory Efficient Attention (O(sqrt(n)) for Jax and PyTorch

A human-readable PyTorch implementation of "Self-attention Does Not Need O(n^2) Memory" (Rabe&Staats'21).

Linear Algebra Fundamentals for Machine Learning

Stable Diffusion with Core ML on Apple Silicon

Reference implementation of the Transformer architecture optimized for Apple Neural Engine (ANE)

Stable Diffusion inference on iOS / macOS using MPSGraph

Implementation of Dreambooth (https://arxiv.org/abs/2208.12242) with Stable Diffusion

Robust Speech Recognition via Large-Scale Weak Supervision

A 3D-printed keyboard inspired by the DataHand

A latent text-to-image diffusion model

Collection of scripts to pretrain T5 in unsupervised text, using PyTorch Lightning. CORD-19 pretraining provided as example.

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

⚡ boost inference speed of T5 models by 5x & reduce the model size by 3x.

Download Walfie GIFs

Code for the paper "Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer"

Pytorch Implementation of EncT5: Fine-tuning T5 Encoder for Non-autoregressive Tasks

min(DALL·E) is a fast, minimal port of DALL·E Mini to PyTorch

WebAssembly JavaScript Interface Creator

Implementation of Denoising Diffusion Probabilistic Model in Pytorch

Karras et al. (2022) diffusion models for PyTorch

scripts for running and training imagen-pytorch

(very personal) Deep learning literature review

High-Resolution Image Synthesis with Latent Diffusion Models