Sea-Snell

Implicit-Language-Q-Learning

Official code from the paper "Offline RL for Natural Language Generation with Implicit Language Q Learning"

JAXSeq

Train very large language models in Jax.

JAX_llama

Inference code for LLaMA models in JAX

grokking

unofficial re-implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"

CALM-Dialogue

Official code for the paper "Context-Aware Language Modeling for Goal-Oriented Dialogue Systems"

micro_config

an opinionated config framework for deep learning based on hierarchical python dataclasses

170project

jax_v_pytorch

comparing Jax (haiku and flax) to PyTorch on simple DL tasks.

AttentionDynamics

visdial-rl

PyTorch code for Learning Cooperative Visual Dialog Agents using Deep Reinforcement Learning

LMTemplate

tpu_pod_launcher

advent_of_code_2023

AirConcierge

AirConcierge-Pytorch-Implementation

AttentionBlogSite

bigcode-evaluation-harness

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

EasyLM

Easy to use model parallel large language models in JAX/Flax with pjit support on cloud TPU pods.

end-to-end-negotiator

Deal or No Deal? End-to-End Learning for Negotiation Dialogues

human-eval

Code for the paper "Evaluating Large Language Models Trained on Code"

jax_diffusion

lm-evaluation-harness

A framework for few-shot evaluation of autoregressive language models.

min_JAX_NERF

re-implementation of NERF in Jax

minima

Minima is a one-size-fits-all Jekyll theme for writers.

mmlu_jaxseq

MMLU evaluation for language models served with EasyLM

perspective

An API wrapper for Google's Perspective API

persuasionforgood

scalax

A simple library for scaling up JAX programs

Sea-Snell.github.io

Personal Site!

SWE-bench

[ICLR 2024] SWE-Bench: Can Language Models Resolve Real-world Github Issues?

transformers

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.