Closed paganpasta closed 2 years ago
An initial idea for the design is:
torch.nn.Linear
eqx.nn.Linear
Open to discussion!
torchvision pretrained support is now added!
torchvision
An initial idea for the design is:
torch.nn.Linear
toeqx.nn.Linear
Open to discussion!