Neural Network Performance Engineer

Join Humanoid, a company on a mission to build the world's most capable humanoid robots. As a Neural Network Performance Engineer, you will work on optimizing neural network-based control policies for smooth robot motions. You will analyze performance bottlenecks, implement custom kernels, and suggest changes to model architecture. The ideal candidate has 3+ years of experience in deep learning systems, 1+ year in neural network inference performance, and strong Python + PyTorch/JAX skills.

• Analyser les goulets d'étranglement de performance d'une architecture de modèle particulière et proposer des améliorations.
• Faire fonctionner le modèle sur un nouveau matériel (par exemple, NVIDIA Thor) de manière efficace.
• Implémenter des noyaux personnalisés pour réduire les exigences de débit mémoire là où cela est nécessaire.