jingyahuang Goto Github PK

accelerate-bert-with-tvm

A benchmark on accelerating BERT based on passes in TVM.

aws-neuron-sdk

Powering AWS purpose-built machine learning chips. Blazing fast and cost effective, natively integrated into PyTorch and TensorFlow and integrated with your favorite AWS services

deep-learning-containers

AWS Deep Learning Containers (DLCs) are a set of Docker images for training and serving models in TensorFlow, TensorFlow 2, PyTorch, and MXNet.

deepspeed

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

ml-training-benchmark

Benchmark for training within different backend.

models

A collection of pre-trained, state-of-the-art models in the ONNX format

neuron-playground

Demos testing neuron (inf1 / inf2 / trn1)

notebooks

Notebooks using the Hugging Face libraries 🤗

onnx

Open standard for machine learning interoperability

onnx-tutorial

This repo gathers examples of ONNX graph surgery.

onnx.github.io

onnxruntime

ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator

onnxruntime-inference-benchmark

onnxruntime-inference-examples

Examples for using ONNX Runtime for machine learning inferencing.

optimum

🏎️ Accelerate training and inference of 🤗 Transformers with easy to use hardware optimization tools

optimum-neuron

Easy, fast and very cheap training and inference on AWS Trainium and Inferentia chips.

ort-playground

Demos testing caveat reported while using onnxruntime

paddlevit

:robot: PaddleViT: State-of-the-art Visual Transformer and MLP Models for PaddlePaddle 2.0+

recurrent-batch-normalization-pytorch

PyTorch implementation of recurrent batch normalization

sagemaker-python-sdk

A library for training and deploying machine learning models on Amazon SageMaker

spleeter

Deezer source separation library including pretrained models.

text-generation-inference

Large Language Model Text Generation Inference

transformers

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

xla

Enabling PyTorch on XLA Devices (e.g. Google TPU)