aj-prime Goto Github PK

deepspeedexamples

Example models using DeepSpeed

dlrm

An implementation of a deep learning recommendation model (DLRM)

elemental

Distributed-memory, arbitrary-precision, dense and sparse-direct linear algebra, conic optimization, and lattice reduction

examples

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.

flexflow

A distributed deep learning framework that supports flexible parallelization strategies.

horovod

Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.

lbann

Livermore Big Artificial Neural Network Toolkit

metaseq

Repo for external large-scale work

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

temp_files

third_party

Third-party source packages that are modified for use in Triton.

triton-server

The Triton Inference Server provides an optimized cloud and edge inferencing solution.