Bernard Lange's starred repositories
datapipelines
Iterable datapipelines for pytorch training.
webdataset
A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.
generative-models
Generative Models by Stability AI
wayve_scenes
Codebase for the WayveScenes101 Dataset
nuscenes-download
script for downloading nuscenes
gpu-optimization-workshop
Slides, notes, and materials for the workshop
llm-course
Course to get into Large Language Models (LLMs) with roadmaps and Colab notebooks.
pytorch-custom-utils
Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new AI research
pytorch_memonger
Experimental ground for optimizing memory of pytorch models
rl-baselines3-zoo
A training framework for Stable Baselines3 reinforcement learning agents, with hyperparameter optimization and pre-trained agents included.
consistency_models
Official repo for consistency models.
x-transformers
A concise but complete full-attention transformer with a set of promising experimental features from various papers
denoising-diffusion-pytorch
Implementation of Denoising Diffusion Probabilistic Model in Pytorch
point-cloud-prediction
Self-supervised Point Cloud Prediction Using 3D Spatio-temporal Convolutional Networks