← all repositories

abhishekkrthakur/tez

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

1.2k stars Python ML Frameworks
tez
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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.