camera ready version

tests/test_activations.py

@@ -1,4 +1,4 @@
-from nanotorch.tensor import Tensor
+from smoltorch.tensor import Tensor
 
 # Test 1: ReLU
 print("Test 1 - ReLU:")

tests/test_add.py

@@ -1,4 +1,4 @@
-from nanotorch.tensor import Tensor
+from smoltorch.tensor import Tensor
 
 # Test 1: Simple addition (no broadcasting)
 a = Tensor([1.0, 2.0, 3.0])

tests/test_linear.py

@@ -1,5 +1,5 @@
-from nanotorch.nn import Linear
-from nanotorch.tensor import Tensor
+from smoltorch.nn import Linear
+from smoltorch.tensor import Tensor
 
 # Test 1: Single sample forward pass
 print("Test 1 - Single forward pass:")

tests/test_matmul.py

@@ -1,4 +1,4 @@
-from nanotorch.tensor import Tensor
+from smoltorch.tensor import Tensor
 
 # Test 1: Simple 2D matmul
 print("Test 1 - Simple 2D matmul:")

tests/test_mlp.py

@@ -1,6 +1,6 @@
 import numpy as np
-from nanotorch.tensor import Tensor
-from nanotorch.nn import MLP
+from smoltorch.tensor import Tensor
+from smoltorch.nn import MLP
 
 # Test 1: MLP forward pass
 print("Test 1 - MLP forward pass:")

tests/test_mul.py

@@ -1,4 +1,4 @@
-from nanotorch.tensor import Tensor
+from smoltorch.tensor import Tensor
 
 # Test 1: Simple multiplication (no broadcasting)
 print("Test 1 - No broadcasting:")

tests/test_reductions.py

@@ -1,4 +1,4 @@
-from nanotorch.tensor import Tensor
+from smoltorch.tensor import Tensor
 
 # Test 1: Sum all elements
 print("Test 1 - Sum (all elements):")