#589 · Primary category: Education & Research

Tensor-Puzzles

machine-learning puzzle torch

Solve puzzles. Improve your pytorch.

Project last updated:07/15/24

GitHub Stars

4.3K

Forks

395

Contributors

3

License

MIT

Why we included this project

Twenty-one puzzles ask you to reimplement a NumPy function using only broadcasting, indexing, and a few allowed operations, each in a single line of code under 80 characters. That constraint forces you to think in terms of shapes and broadcasting rather than reaching for a standard-library shortcut, which is exactly the instinct you need when writing real tensor code. If you build models or process arrays in PyTorch, working through these exercises will make that mental model feel natural. It also works as a self-contained workshop for a team that wants to get more comfortable with NumPy and PyTorch.

Articles for this project

No articles for this project yet.

To suggest a topic or contribute an article, contact us.

Related projects in this category