srush/Tensor-Puzzles: Solve puzzles. Improve your pytorch.

Solve puzzles. Improve your pytorch. Contribute to srush/Tensor-Puzzles development by creating an account on GitHub.

by Sasha Rush - srush_nlp (with Marcos Treviso) When learning a tensor programming language like PyTorch or Numpy it is tempting to rely on the standard library (or more honestly StackOverflow) to find a magic function for everything. But in practice, the tensor language is extremely expressive, and you can do most things from first principles and clever use of broadcasting. This is a collection of 21 tensor puzzles. Like chess puzzles these are not meant to simulate the complexity of a real program, but to practice in a simplified environment. Each puzzle asks you to reimplement one…

saved by

• Jon
• SUJASH AGRAWAL

related reading

• PyTorch internals : ezyang's blogblog.ezyang.com
• Tensor Considered Harmfulnlp.seas.harvard.edu
• the bug that taught me more about PyTorch than years of using it | Elana Simonelanapearl.github.io
• Chapter 0: Fundamentals - ARENAlearn.arena.education
• torch.Tensor — PyTorch 2.12 documentationdocs.pytorch.org
• Tensors — PyTorch Tutorials 2.12.0+cu130 documentationpytorch.org
• Chapter 0: Fundamentals - ARENAlearn.arena.education
• PyTorch (@PyTorch) / Xx.com
• Tensors — PyTorch Tutorials 2.13.0+cu130 documentationdocs.pytorch.org
• Using PyTorch + NumPy? You're making a mistake. · Tanel Pärnamaatanelp.github.io
• torch.unravel_indexdocs.pytorch.org
• Learning PyTorch with Examples — PyTorch Tutorials 2.12.0+cu130 documentationpytorch.org