tocean

tocean's Stars

• Azure/MS-AMP

  Microsoft Automatic Mixed Precision Library

  Language:Python53945

• NVIDIA/TransformerEngine

  A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs, to provide better performance with lower memory utilization in both training and inference.

  Language:Python2.1k339

• pytorch/pytorch

  Tensors and Dynamic neural networks in Python with strong GPU acceleration

  Language:Python85.6k23k

• microsoft/mscclpp

  MSCCL++: A GPU-driven communication stack for scalable AI applications

  Language:C++27643

• Azure/MS-AMP-Examples

  Examples for MS-AMP package.

  Language:Shell2710

• microsoft/hivedscheduler

  Kubernetes Scheduler for Deep Learning

  Language:Go25538

• microsoft/NeuronBlocks

  NLP DNN Toolkit - Building Your NLP DNN Models Like Playing Lego

  Language:Python1.5k195

• microsoft/pai

  Resource scheduling and cluster management for AI

  Language:JavaScript2.6k549

• microsoft/nni

  An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.

  Language:Python14.1k1.8k

• microsoft/vs-tools-for-ai

  Visual Studio Tools for AI is a free Visual Studio extension to build, test, and deploy deep learning / AI solutions. It seamlessly integrates with Azure Machine Learning for robust experimentation capabilities, including but not limited to submitting data preparation and model training jobs transparently to different compute targets. Additionally, it provides support for custom metrics and run history tracking, enabling data science reproducibility and auditing. Enterprise ready collaboration, allow to securely work on project with other people.

  35381