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
561 stars 55 forks source link

Enforce batch axis first #227

Closed schaefertim closed 3 years ago

schaefertim commented 3 years ago

Resolves https://github.com/fKunstner/backpack-discuss/issues/117