Software Engineer V(PyTorch Models)

Duration : 12 Months +
Pay range : $100-$110/HR on W2

• Build/develop exploratory inference system for PyTorch models.
• Analyze/implement performance optimization opportunities running on GPUs or custom HW architectures
• Maintain good engineering practices (unit tests, etc) across the stack

Required Skills

• Fluent with programming in Python, C++
• Strong communication and collaboration skills

Good-to-have Skills

• Experience with PyTorch
• Experience with profiling CPU/GPU or custom HW applications, and identifying performance bottlenecks
• Experience with implementing performance optimizations relevant to CPU/GPU or custom HW applications
• Experience with compilers

Experience

• 10+ YOE as a full-time SWE

Top 3 must-have HARD skills:   

• Python
• C++
• Strong written and verbal communication skills

Good to have skills:

• Experience with PyTorch
• Experience with profiling CPU/GPU or custom HW applications, and identifying performance bottlenecks
• Experience with implementing performance optimizations relevant to CPU/GPU or custom HW applications