Research Associate @ Multimodal Perception Lab-IIIT Bangalore
Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.
Lightning Models is a streamlined toolkit for effortlessly saving, loading, and managing your model checkpoints.
Make PyTorch models up to 40% faster! Thunder is a source to source compiler for PyTorch. It enables using different hardware executors at once; across one or thousands of GPUs.
A PyTorch Implementation Of SimCLR
Torchmetrics - Machine learning metrics for distributed, scalable PyTorch applications.