A modern educational deep learning framework for students, engineers and researchers
Features • Installation • Examples • Documentation • License • Acknowledgments
TensorWeaver is a deep learning framework designed specifically for students, engineers and researchers who want to understand how deep learning frameworks work under the hood. Unlike industrial frameworks like PyTorch and TensorFlow that prioritize performance and scalability, TensorWeaver focuses on clarity, readability, and simplicity.
Built entirely in Python with only NumPy as a dependency, TensorWeaver's codebase is transparent and approachable, making it an ideal learning resource for those who want to demystify the "magic" behind modern AI frameworks.
The target users are students, engineers and researchers who want to fully understand the working principles of deep learning frameworks and gain the skills to debug, extend and optimize their own projects with any deep learning framework.
- Purely Educational: Designed from the ground up as a learning tool with clear, well-documented code
- PyTorch-compatible API: Exactly the same interface as PyTorch, so you can easily switch between TensorWeaver and PyTorch
- Lightweight and Readable: Built with pure Python and minimal dependencies (merely NumPy)
- Fully Functional: Supports essential deep learning components:
- Automatic differentiation engine
- Common neural network operators
- Loss functions and optimizers
- Model definition and training
- Advanced Capabilities:
- ONNX export functionality
- Comprehensive Documentation: Detailed explanations of implementation details and design choices
TensorWeaver is designed to be compatible with PyTorch 2.6, allowing for an easier learning transition between frameworks.
TensorWeaver targets coverage of 20% of the most commonly used PyTorch APIs, following the 80/20 principle (where 20% of features cover 80% of use cases). The specific API list has not been compiled yet and will be determined once appropriate data sources or statistical tools are identified for analyzing API usage patterns.
# Install from PyPI
pip install tensorweaver
# Or install from source
git clone https://github.com/howl-anderson/tensorweaver.git
cd tensorweaver
poetry installsee poetry for more details if you don't have poetry installed.
See milestones for examples.
see https://www.tensorweaver.ai
This project is licensed under the MIT License - see the LICENSE file for details.
- The project draws inspiration from educational frameworks like Micrograd, TinyFlow, and DeZero
- Special thanks to the open-source deep learning community for their pioneering work
- Thanks to all contributors and users who help improve this educational resource