Skip to content

About

A pure-Python subset of the PyTorch API (tensors, extensible autograd) with unit tests and property-based tests. Profiled cache locality, memory usage, and thread scaling to identify performance bottlenecks

Resources

Stars

0 stars

Watchers

0 watching

Forks

 
 

Repository files navigation

A deep learning training tool entirely from scratch using Python, aimed at enabling users to build and train custom deep learning models. The project involved designing efficient data structures to support tensor operations and core deep learning functionalities. Key components included implementing auto-differentiation and backpropagation algorithms, broadcasting mechanisms, and GPU-accelerated parallel programming. The system was optimized for both computational performance and memory usage. Extensive testing was conducted through a suite of custom test cases to ensure the accuracy and reliability of each module.

Additionally you will need to install and download the MNist library.

(On Mac, this may require installing the wget command)

pip install python-mnist
mnist_get_data.sh
  • Tests:
python run_tests.py

About

A pure-Python subset of the PyTorch API (tensors, extensible autograd) with unit tests and property-based tests. Profiled cache locality, memory usage, and thread scaling to identify performance bottlenecks

Resources

Stars

0 stars

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages