tridao
Assistant Professor @ Princeton CS, machine learning & systems
Stanford UniversityPrinceton, NJ
Pinned Repositories
pytorch-lightning
Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.
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
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/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/pytorch-lightning
The lightweight PyTorch wrapper for ML researchers. Scale your models. Write less boilerplate