search

ndif-team / nnsight

The nnsight package enables interpreting and manipulating the internals of deep learned models.
https://nnsight.net/
MIT License
399 stars 37 forks source link

Meta device #178

Closed AdamBelfki3 closed 3 months ago

AdamBelfki3 commented 3 months ago

Loading parameters to the 'meta' device & torch.zeros(), torch.ones(), and torch.rand() are now supported by nnsight via patching.