zhongwen

langchain

🦜🔗 Build context-aware reasoning applications

segment-anything

The repository provides code for running inference with the SegmentAnything Model (SAM), links for downloading the trained model checkpoints, and example notebooks that show how to use the model.

ChatGLM-6B

ChatGLM-6B: An Open Bilingual Dialogue Language Model | 开源双语对话语言模型

Open-Assistant

OpenAssistant is a chat-based assistant that understands tasks, can interact with third-party systems, and retrieve information dynamically to do so.

nanoGPT

The simplest, fastest repository for training/finetuning medium-sized GPTs.

stanford_alpaca

Code and documentation to train Stanford's Alpaca models, and generate the data.

ControlNet

Let us control diffusion models!

JARVIS

JARVIS, a system to connect LLMs with ML community. Paper: https://arxiv.org/pdf/2303.17580.pdf

diffusers

🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.

alpaca-lora

Instruct-tune LLaMA on consumer hardware

StableLM

StableLM: Stability AI Language Models

tiktoken

tiktoken is a fast BPE tokeniser for use with OpenAI's models.

PaLM-rlhf-pytorch

Implementation of RLHF (Reinforcement Learning with Human Feedback) on top of the PaLM architecture. Basically ChatGPT but with PaLM

point-e

Point cloud diffusion for 3D model synthesis

DiT

Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"

LMOps

General technology for enabling AI capabilities w/ LLMs and MLLMs

EasyLM

Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.

natbot

Drive a browser with GPT-3

Llama-X

Open Academic Research on Improving LLaMA to SOTA LLM

maxtext

A simple, performant and scalable Jax LLM!

cramming

Cramming the training of a (BERT-type) language model into limited compute.

diplomacy_cicero

Code for Cicero, an AI agent that plays the game of Diplomacy with open-domain natural language negotiation.

mup

maximal update parametrization (µP)

jax-triton

jax-triton contains integrations between JAX and OpenAI Triton

rlpd

Assemble-Them-All

[SIGGRAPH Asia 2022] Assemble Them All: Physics-Based Planning for Generalizable Assembly by Disassembly

alm

Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective

jax_xc

Exchange correlation functionals translated from libxc to jax

hloenv

an environment based on XLA for deep learning compiler optimization research.

optim4rl

Optim4RL is a Jax framework of learning to optimize for reinforcement learning.