diff --git a/README.md b/README.md index dbf5ba9..d0632b0 100644 --- a/README.md +++ b/README.md @@ -19,8 +19,8 @@ # Deep Learning Framework From Scratch - Unit-tested and documented educational framework. Similar to PyTorch, but with simpler sintax. -- The __autograd engine__ is in [tensor.py](src/tensor.py). I got a lot of inspiration from Andrej Karpathy's __micrograd__ videos. -- The deep learning model __layers__ are in [nn.py](src/nn.py). +- The __autograd engine__ is in [tensor_operations.py](neuralforge/tensor_operations.py). I got a lot of inspiration from Andrej Karpathy's __micrograd__ videos. +- The deep learning model __layers__ are in [nn.py](neuralforge/nn.py).
Check out the implemented basic operations: