kfac-pytorch icon indicating copy to clipboard operation
kfac-pytorch copied to clipboard

Distributed K-FAC Preconditioner for PyTorch

Results 5 kfac-pytorch issues
Sort by recently updated
recently updated
newest added

### Describe the Request Could you add adaptive damping features? Thanks ### Sample Code _No response_

enhancement
help wanted
good first issue

https://github.com/pytorch/pytorch/pull/62303

enhancement
help wanted
pytorch-1.11

https://github.com/pytorch/pytorch/pull/62140 "grouped comm on a set of unflattened tensors can be more performant than flattening+a single flat nccl call."

enhancement
help wanted
pytorch-1.11

enhancement
help wanted