Skip to content

reitwiec/pytorch-mini

 
 

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

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages

  • Python 100.0%