Aidyn-A's repositories
NektarWithScalar
Fork of https://gitlab.nektar.info/nektar/nektar
apex
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
Fuser
A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
lightning-thunder
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.
Megatron-LM
Ongoing research training transformer models at scale
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
pytorch-examples
A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
pytorch-image-models
PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more
pytorch-tutorials
PyTorch tutorials.
torch_ucc
Pytorch process group third-party plugin for UCC
vision
Datasets, Transforms and Models specific to Computer Vision