← all repositories

srush/Tensor-Puzzles

A collection of 21 Jupyter Notebook puzzles that teach PyTorch/NumPy tensor programming through reimplementing library functions from first principles.

4.1k stars Jupyter Notebook Learning
Tensor-Puzzles
Velocity · 7d
+2.7
★ / day
Trend
steady
star history

The repository contains tensor puzzles designed to improve PyTorch skills by practicing broadcasting and tensor operations in a simplified environment. Each puzzle asks users to reimplement one NumPy standard library function in under one line of code, using only specific allowed operations. The project runs in Jupyter Notebooks and includes video walkthroughs for guidance.

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