feat: add linear layer train support#233

Draft

ayoub-louati wants to merge 28 commits intomainfrom

feat/add_linear_layer_train_support

Commits on Jan 10, 2023

Commits on Jan 17, 2023

Commits on Jan 18, 2023

feat: remove debugging print in backward function + fix fp16 tests with autocast and custom_bwd
ayoub-louati
committed

Commits on Jan 20, 2023

feat: fix relu grad + fix benchmark tests
ayoub-louati
committed

Commits on Jan 24, 2023

feat: remove unused import
ayoub-louati
committed

Commits on Jan 27, 2023

Commits on Feb 1, 2023

Merge branch 'main' into feat/add_linear_layer_train_support
ayoub-louati
authored

Commits on Feb 10, 2023

Merge branch 'main' into feat/add_linear_layer_train_support
ayoub-louati
authored