Package exports:
- Fix tinytorch/__init__.py to export all required components for milestones
- Add Dense as alias for Linear for compatibility
- Add loss functions (MSELoss, CrossEntropyLoss, BinaryCrossEntropyLoss)
- Export spatial operations, data loaders, and transformer components
Test infrastructure:
- Create tests/conftest.py to handle path setup
- Create tests/test_utils.py with shared test utilities
- Rename test_progressive_integration.py files to include module number
- Fix syntax errors in test files (spaces in class names)
- Remove stale test file referencing non-existent modules
Documentation:
- Update README.md with correct milestone file names
- Fix milestone requirements to match actual module dependencies
Export system:
- Run tito export --all to regenerate package from source modules
- Ensure all 20 modules are properly exported
Major directory restructure to support both developer and learner workflows:
Structure Changes:
- NEW: src/ directory for Python source files (version controlled)
- Files renamed: tensor.py → 01_tensor.py (matches directory naming)
- All 20 modules moved from modules/ to src/
- CHANGED: modules/ now holds generated notebooks (gitignored)
- Generated from src/*.py using jupytext
- Learners work in notebooks, developers work in Python source
- UNCHANGED: tinytorch/ package (still auto-generated from notebooks)
Workflow: src/*.py → modules/*.ipynb → tinytorch/*.py
Command Updates:
- Updated export command to read from src/ and generate to modules/
- Export flow: discovers modules in src/, converts to notebooks in modules/, exports to tinytorch/
- All 20 modules tested and working
Configuration:
- Updated .gitignore to ignore modules/ directory
- Updated README.md with new three-layer architecture explanation
- Updated export.py source mappings and paths
Benefits:
- Clean separation: developers edit Python, learners use notebooks
- Better version control: only Python source committed, notebooks generated
- Flexible learning: can work in notebooks OR Python source
- Maintains backward compatibility: tinytorch package unchanged
Tested:
- Single module export: tito export 01_tensor ✅
- All modules export: tito export --all ✅
- Package imports: from tinytorch.core.tensor import Tensor ✅
- 20/20 modules successfully converted and exported
Re-exported all modules after restructuring:
- Updated _modidx.py with new module locations
- Removed outdated autogeneration headers
- Updated all core modules (tensor, autograd, layers, etc.)
- Updated optimization modules (quantization, compression, etc.)
- Updated TITO commands for new structure
Changes include:
- 24 tinytorch/ module files
- 24 tito/ command and core files
- Updated references from modules/source/ to modules/
All modules re-exported via nbdev from their new locations.
Added:
- SigmoidBackward class to modules/source/05_autograd/autograd_dev.py with #| export
- BCEBackward class to modules/source/05_autograd/autograd_dev.py with #| export
- Both classes exported to tinytorch/core/autograd.py
- Updated Sigmoid activation to track gradients using SigmoidBackward
- Updated BCE loss to track gradients using BCEBackward
ISSUE: Training still not learning - gradients not flowing properly
- Loss stays constant at 0.7911
- Weights don't update
- Sigmoid.forward() code looks correct but a.requires_grad stays False
- Need to investigate why gradient tracking isn't working through activations
- Created perceptron_trained.py milestone with full training loop
- Restored tinytorch/core/optimizers.py with Optimizer, SGD, Adam, AdamW classes
- Fixed imports to use tinytorch.core.* instead of tensor_dev
- Fixed tinytorch/core/losses.py with all loss functions
- Fixed tinytorch/core/training.py imports
ISSUE: Training loop runs but doesn't learn (gradients not flowing)
- Loss stays constant at 0.7911
- Weights don't update
- Likely autograd (Module 05) backward() not fully implemented
- Need to fix Tensor.backward() and gradient computation
This commit includes:
- Exported tinytorch package files from nbdev (autograd, losses, optimizers, training, etc.)
- Updated activations.py and layers.py with __call__ methods
- New module exports: attention, spatial, tokenization, transformer, etc.
- Removed old _modidx.py file
- Cleanup of duplicate milestone directories
These are the generated package files that correspond to the source modules
we've been developing. Students will import from these when using TinyTorch.
PROBLEM:
- nbdev requires #| export directive on EACH cell to export when using # %% markers
- Cell markers inside class definitions split classes across multiple cells
- Only partial classes were being exported to tinytorch package
- Missing matmul, arithmetic operations, and activation classes in exports
SOLUTION:
1. Removed # %% cell markers INSIDE class definitions (kept classes as single units)
2. Added #| export to imports cell at top of each module
3. Added #| export before each exportable class definition in all 20 modules
4. Added __call__ method to Sigmoid for functional usage
5. Fixed numpy import (moved to module level from __init__)
MODULES FIXED:
- 01_tensor: Tensor class with all operations (matmul, arithmetic, shape ops)
- 02_activations: Sigmoid, ReLU, Tanh, GELU, Softmax classes
- 03_layers: Linear, Dropout classes
- 04_losses: MSELoss, CrossEntropyLoss, BinaryCrossEntropyLoss classes
- 05_autograd: Function, AddBackward, MulBackward, MatmulBackward, SumBackward
- 06_optimizers: Optimizer, SGD, Adam, AdamW classes
- 07_training: CosineSchedule, Trainer classes
- 08_dataloader: Dataset, TensorDataset, DataLoader classes
- 09_spatial: Conv2d, MaxPool2d, AvgPool2d, SimpleCNN classes
- 10-20: All exportable classes in remaining modules
TESTING:
- Test functions use 'if __name__ == "__main__"' guards
- Tests run in notebooks but NOT on import
- Rosenblatt Perceptron milestone working perfectly
RESULT:
✅ All 20 modules export correctly
✅ Perceptron (1957) milestone functional
✅ Clean separation: development (modules/source) vs package (tinytorch)
- Remove circular imports where modules imported from themselves
- Convert tinytorch.core imports to sys.path relative imports
- Only import dependencies that are actually used in each module
- Preserve documentation imports in markdown cells
- Use consistent relative path pattern across all modules
- Remove hardcoded absolute paths in favor of relative imports
Affected modules: 02_activations, 03_layers, 04_losses, 06_optimizers,
07_training, 09_spatial, 12_attention, 17_quantization
Major refactoring:
- Eliminated Variable class completely from autograd module
- Implemented progressive enhancement pattern with enable_autograd()
- All modules now use pure Tensor with requires_grad=True
- PyTorch 2.0 compatible API throughout
- Clean separation: Module 01 has simple Tensor, Module 05 enhances with gradients
- Fixed all imports and references across layers, activations, losses
- Educational clarity: students learn modern patterns from day one
The system now follows the principle: 'One Tensor class to rule them all'
No more confusion between Variable and Tensor - everything is just Tensor!
✅ Fixed all forward dependency violations across modules 3-10
✅ Learning progression now clean: each module uses only previous concepts
Module 3 Activations:
- Removed 25+ autograd/Variable references
- Pure tensor-based activation functions
- Students learn nonlinearity without gradient complexity
Module 4 Layers:
- Removed 15+ autograd references
- Simplified Dense/Linear layers to pure tensor operations
- Clean building blocks without gradient tracking
Module 7 Spatial:
- Simplified 20+ autograd references to basic patterns
- Conv2D/BatchNorm work with basic gradients from Module 6
- Focus on CNN mechanics, not autograd complexity
Module 8 Optimizers:
- Simplified 50+ complex autograd references
- Basic SGD/Adam using simple gradient operations
- Educational focus on optimization math
Module 10 Training:
- Fixed import paths and simplified autograd usage
- Integration module using concepts from Modules 6-9 only
- Clean training loops without advanced patterns
RESULT: Clean learning progression where students only use concepts
they've already learned. No more circular dependencies!
Committing all remaining autograd and training improvements:
- Fixed autograd bias gradient aggregation
- Updated optimizers to preserve parameter shapes
- Enhanced loss functions with Variable support
- Added comprehensive gradient shape tests
This commit preserves the working state before cleaning up
the examples directory structure.
🛡️ **CRITICAL FIXES & PROTECTION SYSTEM**
**Core Variable/Tensor Compatibility Fixes:**
- Fix bias shape corruption in Adam optimizer (CIFAR-10 blocker)
- Add Variable/Tensor compatibility to matmul, ReLU, Softmax, MSE Loss
- Enable proper autograd support with gradient functions
- Resolve broadcasting errors with variable batch sizes
**Student Protection System:**
- Industry-standard file protection (read-only core files)
- Enhanced auto-generated warnings with prominent ASCII-art headers
- Git integration (pre-commit hooks, .gitattributes)
- VSCode editor protection and warnings
- Runtime validation system with import hooks
- Automatic protection during module exports
**CLI Integration:**
- New `tito system protect` command group
- Protection status, validation, and health checks
- Automatic protection enabled during `tito module complete`
- Non-blocking validation with helpful error messages
**Development Workflow:**
- Updated CLAUDE.md with protection guidelines
- Comprehensive validation scripts and health checks
- Clean separation of source vs compiled file editing
- Professional development practices enforcement
**Impact:**
✅ CIFAR-10 training now works reliably with variable batch sizes
✅ Students protected from accidentally breaking core functionality
✅ Professional development workflow with industry-standard practices
✅ Comprehensive testing and validation infrastructure
This enables reliable ML systems training while protecting students
from common mistakes that break the Variable/Tensor compatibility.
BREAKTHROUGH IMPLEMENTATION:
✅ Auto-generated warnings now added to ALL exported files automatically
✅ Clear source file paths shown in every tinytorch/ file header
✅ CLAUDE.md updated with crystal clear rules: tinytorch/ = edit modules/
✅ Export process now runs warnings BEFORE success message
SYSTEMATIC PREVENTION:
- Every exported file shows: AUTOGENERATED! DO NOT EDIT! File to edit: [source]
- THIS FILE IS AUTO-GENERATED FROM SOURCE MODULES - CHANGES WILL BE LOST!
- To modify this code, edit the source file listed above and run: tito module complete
WORKFLOW ENFORCEMENT:
- Golden rule established: If file path contains tinytorch/, DON'T EDIT IT DIRECTLY
- Automatic detection of 16 module mappings from tinytorch/ back to modules/source/
- Post-export processing ensures no exported file lacks protection warning
VALIDATION:
✅ Tested with multiple module exports - warnings added correctly
✅ All tinytorch/core/ files now protected with clear instructions
✅ Source file paths correctly mapped and displayed
This prevents ALL future source/compiled mismatch issues systematically.
- Create professional examples directory showcasing TinyTorch as real ML framework
- Add examples: XOR, MNIST, CIFAR-10, text generation, autograd demo, optimizer comparison
- Fix import paths in exported modules (training.py, dense.py)
- Update training module with autograd integration for loss functions
- Add progressive integration tests for all 16 modules
- Document framework capabilities and usage patterns
This commit establishes the examples gallery that demonstrates TinyTorch
works like PyTorch/TensorFlow, validating the complete framework.
- Regenerate all .ipynb files from fixed .py modules
- Update tinytorch package exports with corrected implementations
- Sync package module index with current 16-module structure
These generated files reflect all the module fixes and ensure consistent
.py ↔ .ipynb conversion with the updated module implementations.
- Export all modules with CIFAR-10 and checkpointing enhancements
- Create demo_cifar10_training.py showing complete pipeline
- Fix module issues preventing clean imports
- Validate all components work together
- Confirm students can achieve 75% CIFAR-10 accuracy goal
Pipeline validated:
✅ CIFAR-10 dataset downloading
✅ Model creation and training
✅ Checkpointing for best models
✅ Evaluation tools
✅ Complete end-to-end workflow
- Exported 09_training module using nbdev directly from Python file
- Exported 08_optimizers module to resolve import dependencies
- All training components now available in tinytorch.core.training:
* MeanSquaredError, CrossEntropyLoss, BinaryCrossEntropyLoss
* Accuracy metric
* Trainer class with complete training orchestration
- All optimizers now available in tinytorch.core.optimizers:
* SGD, Adam optimizers
* StepLR learning rate scheduler
- All components properly exported and functional
- Integration tests passing (17/17)
- Inline tests passing (6/6)
- tito CLI integration working correctly
Package exports:
- tinytorch.core.training: 688 lines, 5 main classes
- tinytorch.core.optimizers: 17,396 bytes, complete optimizer suite
- Clean separation of development vs package code
- Ready for production use and further development
