https://github.com/srush/tensor-puzzles-penzai
https://github.com/srush/tensor-puzzles-penzai
Last synced: 5 months ago
JSON representation
- Host: GitHub
- URL: https://github.com/srush/tensor-puzzles-penzai
- Owner: srush
- Created: 2024-04-21T22:38:39.000Z (about 2 years ago)
- Default Branch: main
- Last Pushed: 2024-04-22T13:32:31.000Z (about 2 years ago)
- Last Synced: 2025-01-28T05:41:18.324Z (over 1 year ago)
- Language: HTML
- Size: 6.47 MB
- Stars: 20
- Watchers: 2
- Forks: 1
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# Tensor Puzzles Penzai
This is an implementation of the [Tensor Puzzles](https://github.com/srush/Tensor-Puzzles) written in the new [Penzai](https://penzai.readthedocs.io/en/stable/index.html) Jax framework.
It is meant to test out both the automatic visualization and named tensor support of the library.