Try to define new pytorch functions and modules
- functions.py - where new functions defined
- modules.py - where new modules defined
- reference.py - reference MLP implementation (copy from here)
- custom_function.py - MLP using customized ReLU function
- custom_module.py - simple MLP using customized modules (ReLU, Linear)