mirror of
https://github.com/MLSysBook/TinyTorch.git
synced 2026-05-31 00:07:37 -05:00
302cbea5ff
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.
15 lines
464 B
Python
Generated
15 lines
464 B
Python
Generated
# AUTOGENERATED! DO NOT EDIT! File to edit: ../../modules/source/06_optimizers/optimizers_dev.ipynb.
|
|
|
|
# %% auto 0
|
|
__all__ = []
|
|
|
|
# %% ../../modules/source/06_optimizers/optimizers_dev.ipynb 1
|
|
import numpy as np
|
|
from typing import List, Union, Optional, Dict, Any
|
|
|
|
# Import Tensor from Module 01 (now with gradient support from Module 05)
|
|
import sys
|
|
import os
|
|
sys.path.append(os.path.join(os.path.dirname(__file__), '..', '01_tensor'))
|
|
from tensor_dev import Tensor
|