Become an AI Compiler Software Developer at yasp, where you’ll enhance AI training through intelligent compiler components. Work remotely alongside top ML practitioners.
As part of yasp's compiler team, your expertise in Python and C/C++ will be critical for optimizing deep learning models. In this role, you will collaborate with compiler and ML specialists to automate code generation and improve performance across a range of hardware solutions. This position focuses on developing innovative features that advance AI infrastructure.
Key Responsibilities:
• Collaborate on designing and optimizing AI workload processes
• Automate kernel optimization using LLM-based agents
• Implement complex compiler passes for AI workloads
• Identify and resolve performance issues in training pipelines
• Create tailored optimizations for backend AI targets
Requirements:
• Solid skills in Python and C/C++ programming
• Experience with frameworks such as ONNX or TensorRT
• ...