Senior Software Engineer — cuEquivariance

What you’d do

Build and optimize CUDA kernels for equivariant neural network primitives, and maintain PyTorch and JAX interfaces exposing them. Full time role covering implementation through production delivery, CI/CD, and collaboration with research teams. Found on the employer's careers page on Sep 29, 2026. Apply through the link.