Software Engineer - GPU Kernel - Remote

Overview

This role involves working with Mako's AI platform to optimize GPU performance for generative AI models, focusing on kernel tuning and system optimizations.

In Short

• Join the R&D team at Mako to enhance AI model deployment efficiency.
• Lead improvements in GPU performance through kernel optimization.
• Work with technologies like CUDA, ROCm, and Triton.
• Analyze performance bottlenecks in ML training and inference.
• Implement solutions in C/C++ and Python.
• Collaborate on extending machine learning compilers and frameworks.
• Engage in professional development opportunities.
• Enjoy flexible working hours and remote work options.
• Receive competitive salary and performance-based bonuses.
• Participate in team-building events and social activities.

Requirements

• Bachelor's, Master’s, or PhD in Computer Science or related field.
• Strong programming skills in C/C++ and Python.
• Experience in GPU performance optimizations.
• Proven experience with kernel optimizations on CUDA or ROCm.
• General experience with ML model training and deployment.
• Experience with distributed systems development.

Benefits

• Competitive salary package.
• Performance-based bonuses and incentives.
• Comprehensive health insurance coverage.
• Flexible working hours and remote work options.
• Professional development opportunities.
• Generous vacation and paid time off policy.
• Company-sponsored social activities.
• Modern work environment with state-of-the-art equipment.