Skip to content
This repository was archived by the owner on Jan 28, 2026. It is now read-only.

feat: add linear layer train support#233

Draft
ayoub-louati wants to merge 28 commits intomainfrom
feat/add_linear_layer_train_support
Draft

feat: add linear layer train support#233
ayoub-louati wants to merge 28 commits intomainfrom
feat/add_linear_layer_train_support

Conversation

@ayoub-louati
Copy link
Contributor

@ayoub-louati ayoub-louati commented Jan 9, 2023

Add training support for linear layer:
closes #228

@ayoub-louati ayoub-louati self-assigned this Jan 9, 2023
@ayoub-louati ayoub-louati marked this pull request as ready for review January 12, 2023 09:02
@ayoub-louati ayoub-louati changed the title Feat/add linear layer train support feat: add linear layer train support Jan 12, 2023
@pommedeterresautee pommedeterresautee removed the request for review from gaetansnl January 19, 2023 09:19
@pommedeterresautee pommedeterresautee marked this pull request as draft January 19, 2023 09:19
@ayoub-louati ayoub-louati marked this pull request as ready for review January 23, 2023 20:17
@ayoub-louati
Copy link
Contributor Author

ayoub-louati commented Jan 23, 2023

@pommedeterresautee all tests (backward benchmark) passed on cloud-dev (without cuda-graphs):
image
All tests (foraward and backward benchmakrs) passed:
image

@ayoub-louati ayoub-louati marked this pull request as draft February 10, 2023 09:27
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

Development

Successfully merging this pull request may close these issues.

linear layer backward

1 participant