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

feat: add bw pass on layernorm/rmsnorm#164

Draft
pommedeterresautee wants to merge 4 commits intomainfrom
feat/backward_layernorm
Draft

feat: add bw pass on layernorm/rmsnorm#164
pommedeterresautee wants to merge 4 commits intomainfrom
feat/backward_layernorm

Conversation

@pommedeterresautee
Copy link
Member

No description provided.

@pommedeterresautee pommedeterresautee self-assigned this Nov 7, 2022
@samhavens
Copy link

I notice this PR has been open for a while; what's its status?

@pommedeterresautee
Copy link
Member Author

@samhavens hi, no real issue, just focusing on inference right now... Is it something you are interested into?

@samhavens
Copy link

Not having to deal with CUDA for an RMSnorm kernel is appealing, yeah 😄 it's not high priority currently, but I wanted to make sure to keep tabs on this PR 👀

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.

2 participants