pytorch-fx topic

List pytorch-fx repositories

opt_einsum_fx

18
Stars
5
Forks
Watchers

Einsum optimization using opt_einsum and PyTorch FX graph rewriting

ml-tune-mode-convbn

15
Stars
0
Forks
Watchers

Tune-Mode ConvBN Blocks For Efficient Transfer Learning