tensorweaver 0.0.3

A modern educational deep learning framework for students, engineers and researchers

TensorWeaver

A modern educational deep learning framework for students, engineers and researchers

Features • Installation • Examples • Documentation • License • Acknowledgments

Introduction

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.

Features

• 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

PyTorch Compatibility

TensorWeaver is designed to be compatible with PyTorch 2.6, allowing for an easier learning transition between frameworks.

PyTorch API Coverage

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.

Online demo

Click here to launch the online demo

Installation

# Install from PyPI
pip install tensorweaver

# Or install from source
git clone https://github.com/howl-anderson/tensorweaver.git
cd tensorweaver
poetry install

see poetry for more details if you don't have poetry installed.

Examples

See milestones for examples.

Documentation

see https://www.tensorweaver.ai

License

This project is licensed under the MIT License - see the LICENSE file for details.

Acknowledgments

• 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