LunjunZhang
Follow
Stars
The simplest, fastest repository for training/finetuning medium-sized GPTs.
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
A community-maintained Python framework for creating mathematical animations.
Python Fire is a library for automatically generating command line interfaces (CLIs) from absolutely any Python object.
Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch
A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training
Official inference repo for FLUX.1 models
DALL·E Mini - Generate images from a text prompt
End-to-End Object Detection with Transformers
Machine Learning Engineering Open Book
PyTorch implementation of MAE https//arxiv.org/abs/2111.06377
A Neural Net Training Interface on TensorFlow, with focus on speed + flexibility
Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.
Tools for merging pretrained large language models.
A concise but complete full-attention transformer with a set of promising experimental features from various papers
Official implementations for paper: Anydoor: zero-shot object-level image customization
PyTorch implementation of Advantage Actor Critic (A2C), Proximal Policy Optimization (PPO), Scalable trust-region method for deep reinforcement learning using Kronecker-factored approximation (ACKT…
A library of reinforcement learning components and agents
A deep learning library for video understanding research.
Official repo for VGen: a holistic video generation ecosystem for video generation building on diffusion models
Make huge neural nets fit in memory