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