yanboliang Goto Github PK

benchmark

TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.

bound-optimization

Linear regression and logistic regression under bound constrained optimization in Python.

einops

Deep learning operations reinvented (for pytorch, tensorflow, jax and others)

gpt-fast

Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.

keras

Deep Learning for humans

megatron-lm

Ongoing research training transformer models at scale

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

pytorch-jit-paritybench

spark

Mirror of Apache Spark

spark-vlbfgs

Vector-free L-BFGS implementation for Spark MLlib

tensorflow

Computation using data flow graphs for scalable machine learning

torchdynamo

A Python-level JIT compiler designed to make unmodified PyTorch programs faster.

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.

triton

Development repository for the Triton language and compiler