tridao
Assistant Professor @ Princeton CS, machine learning & systems
Stanford UniversityPrinceton, NJ
Pinned Repositories
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Augmentation
Data augmentation
cvxpy
A Python-embedded modeling language for convex optimization problems.
flash-attention-wheels
leastsquares
CS 240H final project
metal
Snorkel MeTaL: A framework for training models with multi-task weak supervision
promptsource
Toolkit for creating, sharing and using natural language prompts.
pytorch-lightning
The lightweight PyTorch wrapper for ML researchers. Scale your models. Write less boilerplate
trlx
A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)
zoo
tridao's Repositories
tridao/flash-attention-wheels
tridao/zoo
tridao/leastsquares
CS 240H final project
tridao/trlx
A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)
tridao/pytorch-lightning
The lightweight PyTorch wrapper for ML researchers. Scale your models. Write less boilerplate
tridao/Augmentation
Data augmentation
tridao/cvxpy
A Python-embedded modeling language for convex optimization problems.
tridao/metal
Snorkel MeTaL: A framework for training models with multi-task weak supervision
tridao/promptsource
Toolkit for creating, sharing and using natural language prompts.