onnxruntime-training-examples

Transformer accelerator

Accelerates training of large transformer models by providing optimized kernels and memory optimizations.

Examples for using ONNX Runtime for model training.

GitHub

317 stars
44 watching
61 forks
Language: C#
last commit: almost 2 years ago

Related projects:

RepositoryDescriptionStars
microsoft/onnxruntime-inference-examplesRepository providing examples for using ONNX Runtime (ORT) to perform machine learning inferencing.1,243
microsoft/onnxruntimeA cross-platform, high-performance machine learning accelerator14,990
microsoft/megatron-deepspeedResearch tool for training large transformer language models at scale1,926
alrevuelta/connxrAn embedded device-friendly C ONNX runtime with zero dependencies196
bigscience-workshop/megatron-deepspeedA collection of tools and scripts for training large transformer language models at scale1,342
emergentorder/onnx-scalaAn API and backend for running ONNX models in Scala 3 using typeful, functional deep learning and classical machine learning.138
hermanussen/compiletimemethodexecutiongeneratorAllows executing code during compilation to improve runtime performance20
openai/finetune-transformer-lmThis project provides code and model for improving language understanding through generative pre-training using a transformer-based architecture.2,167
microsoft/mpnetDevelops a method for pre-training language understanding models by combining masked and permuted techniques, and provides code for implementation and fine-tuning.288
uber/neuropodA unified interface to run deep learning models from multiple frameworks using C++ and Python.937
mit-han-lab/data-efficient-gansImproves GAN training efficiency by incorporating data augmentation1,286
german-nlp-group/german-transformer-trainingTrains German transformer models to improve language understanding23
fastnlp/cptA pre-trained transformer model for natural language understanding and generation tasks in Chinese482
kraiskil/onnx2cGenerates C code from ONNX files for efficient neural network inference on microcontrollers234
soumith/imagenet-multigpu.torchA toolkit for training neural networks on the ImageNet dataset using multiple GPUs.402