Tri Dao

No description.

No description.

CS 240H final project

CUDA Templates and Python DSLs for High-Performance Linear Algebra

The lightweight PyTorch wrapper for ML researchers. Scale your models. Write less boilerplate

A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)

Toolkit for creating, sharing and using natural language prompts.

A Python-embedded modeling language for convex optimization problems.

Snorkel MeTaL: A framework for training models with multi-task weak supervision

Data augmentation