-
@google Cloud
- Santa Clara, CA
Starred repositories
A JAX research toolkit for building, editing, and visualizing neural networks.
This repo implements a Stable Diffusion model in PyTorch with all the essential components.
Measuring Massive Multitask Language Understanding | ICLR 2021
JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
TFRecorder makes it easy to create TensorFlow records (TFRecords) from Pandas DataFrames and CSVs files containing images or structured data.
FinRL: Financial Reinforcement Learning. π₯
Taming Transformers for High-Resolution Image Synthesis
Trax β Deep Learning with Clear Code and Speed
Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.
Ultimate Python study guide for newcomers and professionals alike. π π π
Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.
A small package to create visualizations of PyTorch execution graphs
Neural Turing Machines (NTM) - PyTorch Implementation
An easy-to-follow Pytorch implementation of Hinton's Capsule Network
Kubernetes IN Docker - local clusters for testing Kubernetes
tensorboard for pytorch (and chainer, mxnet, numpy, ...)
Open Source Neural Machine Translation and (Large) Language Models in PyTorch
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (Vβ¦
Automated provisioner of a Google Cloud TPU environment for training in PyTorch
Advanced Deep Learning and Reinforcement Learning course taught at UCL in partnership with Deepmind
π€ Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
A helpful micro-framework for writing Kubernetes Admission Controllers ππ