irlyngaas's repositories
apex
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
batchgenerators
A framework for data augmentation for 2D and 3D image classification and segmentation
brain_mask
3D deep convolutional neural network based segmentation and brain masking tool for multi-modal brain MRI
composable_kernel
Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators
flash-attention
Fast and memory-efficient exact attention
Megatron-LM
Ongoing research training transformer language models at scale, including: BERT & GPT-2
miniWeather
A parallel programming training mini app simulating weather-like flows
MPAS-Release
Repository for MPAS models and shared framework releases.
onnxruntime
ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator