Senior Machine Learning Framework Software (México)

Remote Mexico, MexicoPositionC/C++ – Automotive IndustryProject DescriptionWorking on GPU support for OpenAI/Triton — a language and compiler for writing highly efficient custom Deep-Learning primitives.Work with the open‐source community to analyze, develop, test, and deploy performance improvements for neural networks implemented with Triton on GPUs with ROCm.Responsibilities- New features development, support and optimization of OpenAI/Triton project for GPUs.
- Communication with other developers, customers and project managers.
- Test implementation, project documentation and verification of system with unit/component/functional tests.Required Skills- Strong C/C++ programming skills.
- Basic Python programming skills.
- Experience in performance analysis.
- Experience with GPGPU computing (HIP, CUDA, OpenCL, etc.) or compiler internals (LLVM, GCC, or any other).
Nice to Have Skills- Basic understanding of ML technologies.
- Experience with PyTorch.