Senior Machine Learning Framework Software Development Engineer

Position
Senior MachineLearning Framework Software Development Engineer,
Remote Mexico

Develop, support and optimize PyTorch project for GPU. Communication with developers, customers, and project managers. Create, document, and execute unit/component/functional tests. This might involve one or more of the following:
• Modifying PyTorch UT code to adapt it for ROCm
• Filing JIRAs on ROCm components that have missing features needed for the UTs

Must have

• Proficiency with PyTorch, С/C++ and Python
• Knowledge of GPU computing (HIP or CUDA or OpenCL)
• Experience in CMake, make/ninja build system, docker
• Software Version Control (Git)
• Knowledge in ML/NN
• GDB debugging

The successful candidate should possess the following traits:
• Proficiency in problem-solving and troubleshooting technical issues
• Time management, prioritization and multi-tasking skills
• Ability to handle multiple competing priorities in a fast-paced environment
• Personal initiative, commitment, perseverance and resilience
• Well-developed communication and teamwork skills
• Aspiration to engineering excellence

Nice to have

• ROCm
• HIP
• Jenkins
• CUDA
• Pytorch internals

Remote Mexico