facebookresearch / ToMe

A method to increase the speed and lower the memory footprint of existing vision transformers.
Other
931 stars 67 forks source link

Add CUDA extensions #41

Closed vinnamkim closed 2 months ago