Mountagha / Module-3

Module 3 - Efficiency

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

MiniTorch Module 3

MiniTorch is a teaching library for machine learning engineers who wish to learn about the internal concepts underlying deep learning systems. Specifically, it is a pure Python re-implementation of the Torch API designed to be simple, easy-to-read, tested, and incremental. The final library can run Torch code with minimal changes (at some efficiency cost). The project was developed for the course Machine Learning Engineering at Cornell Tech.

The project is organized into 5 modules. Each module is build upon the precedents.

This is Module-3 and it is focus on taking advantage of tensors to write fast code, first on standars CPUs and then GPU.

Notes

This module requires scalar.py, tensor_functions.py, tensor_data.py, tensor_ops.py, operators.py, module.py, and autodiff.py from Module 2.

You will need to modify tensor_functions.py slightly in this assignment.

  • Tests:
python run_tests.py
  • Note:

Several of the tests for this assignment will only run if you are on a GPU machine and will not run on github's test infrastructure. Please follow the instructions to setup up a colab machine to run these tests.

About

Module 3 - Efficiency


Languages

Language:Python 100.0%