Senior Machine Learning Framework Software Development Engineer, Remote Brazil

??Important: Please apply with your CV in English.????Luxoft only operates under CLT contracts.?? Project Description: Working 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. Skills Must have Strong C/C++ programming skills Basic Python programming skills Experience in performance analysis Experience with GPGPU (General purpose GPU) computing (HIP, CUDA, OpenCL, etc.) or Experience with compiler internals (llvm, gcc or any other) Nice to have Basic understanding of ML technologies Experience with PyTorch Experience with LLVM and MLIR compiler infrastructure, analysis or optimizations implementation Knowledge of ROCm infrastructure Experience in CMake, make/ninja build system GEMM performance fundamentals Experience with Docker Languages: English: C1 Advanced