search

f-dangel / backpack

BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.
https://backpack.pt/
MIT License
555 stars 55 forks source link

[DOC] Add RNN example #229

Closed schaefertim closed 2 years ago

schaefertim commented 2 years ago

Adds an example that demonstrates how to use RNNs with BackPACK (through custom modules or the converter).