Build, implement, and optimize CUDA kernels for equivariant neural network primitives — tensor products, segmented polynomials, and triangle-based operations — targeting peak performance across NVIDIA GPU generations.
- 6+ years experience
- Degree or equivalent experience
- Apply on company site
- Benefits
- Equity