TM's repositories
nccl-tests
NCCL Tests without MPI
interactive-decision-tree
Data visualization project that helps preprocess and analyze small data set.
cythonize-setuptools
Cythonize python code and build wheel distribution package
DALI
A library containing both highly optimized building blocks and an execution engine for data pre-processing in deep learning applications
incubator-tvm
Open deep learning compiler stack for cpu, gpu and specialized accelerators
ray
A system for parallel and distributed Python that unifies the ML ecosystem.
tensorflow
An Open Source Machine Learning Framework for Everyone
llm-awq
[MLSys 2024 Best Paper Award] AWQ: Activation-aware Weight Quantization for LLM Compression and Acceleration
Megatron-LM
Ongoing research training transformer models at scale
Text-Classification-PyTorch
Implementation of papers for text classification task on SST-1/SST-2
TurboTransformers
a fast and user-friendly runtime for transformer inference on CPU and GPU