Posts by Alexandre Milesi
Technical Walkthrough
Aug 24, 2021
Accelerating SE(3)-Transformers Training Using an NVIDIA Open-Source Model Implementation
SE(3)-Transformers are versatile graph neural networks. NVIDIA just released an open-source optimized implementation that uses 9x less memory and is up to 21x faster than the baseline official impleme...
8 MIN READ