3 contributions
lightning-ai/pytorch-lightningContributor

Deep learning framework to train, deploy, and scale PyTorch models fast.

Key Contributions:Contributed PyTorch compiler optimizations and CUDA device-level bug fixes.
Python28.1k3.4k
View Repo
lightning-ai/litdataContributor

Streamline data pipelines for AI. Process datasets across 1000s of machines, and optimize data for blazing fast model training.

Key Contributions:Improved performance on distributed data stream loading and multi-GPU staging.
Python34248
View Repo
pytorch/pytorchContributor

Tensors and Dynamic neural networks in Python with strong GPU acceleration.

Key Contributions:Helped optimize compiler frontend, CUDA graphs execution paths, and memory optimizations.
C++ / Python74.5k20.1k
View Repo