From 777513bc31d2838dc1706562cac71bc2469ecb60 Mon Sep 17 00:00:00 2001
From: Siddesh Sambasivam <plutocrat45@gmail.com>
Date: Sun, 27 Jun 2021 22:10:30 +0530
Subject: [PATCH] add todos

---
 matterix/loss_fn.py |  1 +
 matterix/nn.py      | 10 ++++++++--
 matterix/tensor.py  |  2 +-
 3 files changed, 10 insertions(+), 3 deletions(-)

diff --git a/matterix/loss_fn.py b/matterix/loss_fn.py
index e69de29..4d0197b 100644
--- a/matterix/loss_fn.py
+++ b/matterix/loss_fn.py
@@ -0,0 +1 @@
+# TODO: MSE, RMSE, MAE, Binary cross-entropy, Categorical cross-entropy, kullback leibler divergence loss
diff --git a/matterix/nn.py b/matterix/nn.py
index 3e1dcef..fe47ffb 100644
--- a/matterix/nn.py
+++ b/matterix/nn.py
@@ -1,3 +1,9 @@
 # TODO: Module class to represent neural networks
-# TODO: Loss functions
-# TODO: Optimizers
+
+
+class Module:
+    def __init__(self) -> None:
+        pass
+
+    def zero_grad(self) -> None:
+        pass
diff --git a/matterix/tensor.py b/matterix/tensor.py
index 1894e67..738e316 100644
--- a/matterix/tensor.py
+++ b/matterix/tensor.py
@@ -4,7 +4,7 @@
 ArrayableType = Union[float, list, np.ndarray]
 TensorableType = Union[float, np.ndarray, "Tensor"]
 
-# TODO: randn, normal, randint
+# TODO: randn, normal, randint, argmax
 
 
 def enforceTensor(_input: TensorableType) -> "Tensor":