Fast lexical search implementing BM25 in Python using Numpy, Numba and Scipy

A flexible and efficient codebase for training visually-conditioned language models (VLMs)

Retrieval and Retrieval-augmented LLMs

CLIP-like model evaluation

Run Effective Large Batch Contrastive Learning Beyond GPU/TPU Memory Constraint

The fast python bm25 algorithm implemented with reverted index

Temperature Schedules for self-supervised contrastive methods on long-tail data (ICLR'23)

What the f*ck Python? 😱

pfevaluator: A library for evaluating performance metrics of Pareto fronts in multiple/many objective optimization problems

Estimate/count FLOPS for a given neural network using pytorch

Tree edit distance using the Zhang Shasha algorithm

Code for the paper titled "Recursive Top-Down Production for Sentence Generation with Latent Trees"

Simple, Elegant, Typed Argument Parsing with argparse

Pytorch library for fast transformer implementations

Ladder Variational Autoencoders (LVAE) in PyTorch

Vector Quantized VAEs - PyTorch Implementation

PyTorch code to run synthetic experiments.

Pytorch implementation of Hyperspherical Variational Auto-Encoders

A glorified personal dot files

Simple language-driven navigation tasks for studying compositional learning

A LARS implementation in PyTorch

Compositional Obverter Communication Learning From Raw Visual Input - Pytorch Implementation

The Synbols dataset generator is a ServiceNow Research project that was started at Element AI.

📊 Save matplotlib figures as TikZ/PGFplots for smooth integration into LaTeX.

StyleGAN2 - Official TensorFlow Implementation

Benzina is an image-loader package that greatly accelerates image loading onto GPUs using their built-in hardware codecs.

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

The author's officially unofficial PyTorch BigGAN implementation.

WaveGAN: Learn to synthesize raw audio with generative adversarial networks

A machine learning library for PyTorch