abhishekkrthakur/tez
A lightweight PyTorch trainer with utilities for deep learning projects supporting CPU, GPU, and TPU training.

Velocity · 7d
+0.6
★ / day
Trend
→steady
star history
Tez is a minimal PyTorch training library that simplifies deep learning model development. It wraps common training loop patterns while keeping users close to raw PyTorch, supporting single and distributed training across CPU, GPU, and TPU. The library aims to reduce boilerplate code for prototyping and production deployments.