GAO WEI's repositories
ATSPrivacy
AutoML, Privacy Preserving, Federated Learning
ZeroShotLinearStyleTransfer
Arbitrary Style Transfer
annotated_latex_equations
Examples of how to create colorful, annotated equations in Latex using Tikz.
cogdl
CogDL: An Extensive Toolkit for Deep Learning on Graphs
examples
A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
HeliosData
Helios Traces in SenseTime
hfta
Boost hardware utilization for ML training workloads via Inter-model Horizontal Fusion
hire-cluster-simulator
Switches for HIRE: Resource Scheduling for Data Center In-Network Computing
k8s-device-plugin
NVIDIA device plugin for Kubernetes
mlflow
Open source platform for the machine learning lifecycle
mmdetection
Open MMLab Detection Toolbox and Benchmark
sagemaker-debugger
Amazon SageMaker Debugger provides functionality to save tensors during training of machine learning jobs and analyze those tensors
SelfTune
SelfTune is an RL framework that enables systems and service developers to automatically tune various configuration parameters and other heuristics in their codebase, rather than manually-tweaking, over time in deployment. It provides easy-to-use API (Python, C# bindings) and is driven by bandit-style RL & online gradient-descent algorithms.
TitanSched
DL Scheduler; Foundation Models; DL training
triton
Development repository for the Triton language and compiler
vHive
vHive: Open-source framework for serverless experimentation