OhadRubin

scikit-llm

Seamlessly integrate LLMs into scikit-learn.

annoy

Approximate Nearest Neighbors in C++/Python optimized for memory usage and loading/saving to disk

awesome-langchain

😎 Awesome list of tools and projects with the awesome LangChain framework

open_flamingo

An open-source framework for training large multimodal models.

BIG-bench

Beyond the Imitation Game collaborative benchmark for measuring and extrapolating the capabilities of language models

bigcode-evaluation-harness

A framework for the evaluation of autoregressive code generation language models.

UDR

ACL'23: Unified Demonstration Retriever for In-Context Learning

easy-lora-and-gptq

JAX notebook showing how to LoRA + GPTQ arbitrary models

langchain

🦜🔗 Build context-aware reasoning applications

news-please

news-please - an integrated web crawler and information extractor for news that just works

OpenSubtitles

A series of scripts to download and parse the OpenSubtitles corpus.

babyagi

tputils

Utilities for TPUs

webby

Web utilities - HTTP headers, query parsing etc

rvq-vae-gpt

My attempts at applying Soundstream design on learned tokenization of text and then applying hierarchical attention to text generation

Poe

Reverse engineered API for Quora's Poe - access Claude (Anthropic) and ChatGPT (OpenAI)

simple-pytree

A dead simple Python package for creating custom JAX pytree objects

k-diffusion

Karras et al. (2022) diffusion models for PyTorch

levanter

Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax

jax-triton

jax-triton contains integrations between JAX and OpenAI Triton

in-context-ralm

dreamerv2_JAX

An implementation of DreamerV2 written in JAX, with support for running multiple random seeds of an experiment on a single GPU.

jax-smi

JAX Synergistic Memory Inspector

tuning_playbook

A playbook for systematically maximizing the performance of deep learning models.

WeightWatcher

The WeightWatcher tool for predicting the accuracy of Deep Neural Networks

youtube-dl

Command-line program to download videos from YouTube.com and other video sites

latent-browser

Latent web browser

overleaf

A web-based collaborative LaTeX editor

excalidraw

Virtual whiteboard for sketching hand-drawn like diagrams

jax-vqvae-gpt

Implementation of VQ-VAE with a GPT-style sampler in the JAX and Haiku ecosystem.