Multimodal computer agent data collection program

Implementation of Block Recurrent Transformer - Pytorch

Model parallel transformers in JAX and Haiku

Easily turn large sets of image urls to an image dataset. Can download, resize and package 100M urls in 20h on one machine.

jax version of clip guided diffusion scripts

Official repository with code and data accompanying the NAACL 2021 paper "Hurdles to Progress in Long-form Question Answering" (https://arxiv.org/abs/2103.06332).

Very deep VAEs in JAX/Flax

Implementation of Marge, Pre-training via Paraphrasing, in Pytorch