Brandon Trabucco

da-fusion. Effective Data Augmentation With Diffusion Models

design-bench. Benchmarks for Model-Based Optimization

design-baselines. Baselines for Model-Based Optimization

lstm-cuda. This is a c++ implementation of an LSTM Neural Network parallelized for a GPU using CUDA

nerf. Neural Radiance Fields in PyTorch

vq-vae-2. Vector Quantized Latent Variable Models In TF 2.0

program-gan. We propose an architecture using the recent Generative Adversarial Networks and Hyper Networks to generate syntactically correct and behaviorally useful python source code from erroneous existing code.

mass. A Simple Approach For Visual Room Rearrangement: 3D Mapping & Semantic Search (ICLR 2023)

scaled_dot_product_attention. Implements the scaled dot product attention mechamism from the transformer. Vaswani, A. et al. https://arxiv.org/pdf/1706.03762.pdf

pixelcnn. Conditional Gated Pixel CNN in TensorFlow 2

transformer_capsule_layer. Implements a convolutional capsule layer with a scaled dot product attention mechanism.

detailed_captioning. Implements state-of-the-art image captioning algorithms, including the bottom-up top-down attention paper.

bvn. Birkhoff-von Neumann Decomposition with a Greedy Birkhoff Heuristic

tensorflow-hungarian. Guide for building custom op for TensorFlow

insta-dev. Official Development Repo for InSTA

constrained_beam_search. A tensorflow implementation of constrained beam search that supports arbitrary constraint definitions.

morphing-agents. Collection Of Dynamic Morphology Agents For MuJoCo

mineral. A minimalist reinforcement learning package for TensorFlow 2.0

glove. An implementation of a loader for the GloVe word embeddings.

playground. Deep Reinforcement Learning

deepfashion_dataset. TensorFlow build script for the deep fashion dataset, depends on glove embeddings.

off-policy. Off-Policy Reinforcement Learning Algorithms

hypercomplex-ops. Utility functions for TensorFlow that enable working with differentiable hypercomplex numbers.

cider. A pure python cider scorer using numpy, see: https://arxiv.org/abs/1411.5726.

spork. CLI For Experiments On Slurm / Singularity

probabilistic_vq. Implements probabilistic vector quantization for training discrete VAEs

separable_attention. Implements a multi head attention mechanism for images.

best_first. Non-Sequential Decoding Strategies for Image Captioning

wikipedia_dataset. This is a repository using the Wiki Extractor to build and prepare WIKIPEDIA for use in tensorflow.

gesture. A Java based gesture recognition app that uses a Convolutional LSTM Neural Network

im2txt_match. This is an augmentation of the Show and Tell model trained with an image caption pairwise discriminator.