Pinned Repositories
pytorch-lightning
Pretrain, finetune and deploy AI models 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/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
tridao/trlx
A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)