Senior Software Engineer

Summary

The job is for a Machine Learning Engineer at CentML, a company specializing in AI and ML. The role involves optimizing GPU kernels to improve deep learning performance. The candidate should have a degree in Computer Science or Engineering, strong coding skills, and experience with GPU architecture and GPGPU programming.

Requirements

• Bachelor's degree in Computer Science or Engineering
• Excellent communication skills and ability to work in a team
• Strong coding skills in at least one of Python and C++
• Solid fundamentals in other computer science and computer engineering topics

Responsibilities

Communicate with product teams to understand problems, profile ML/DL workloads, find bottlenecks, optimize GPU kernels, write tests and benchmarks

Preferred Qualifications

• 5+ years of experience in researching or contributing to HPC/ML/DL systems, frameworks or libraries
• Experience with GPU architecture and GPGPU programming (NVIDIA GPUs, AMD GPUs, OpenCL)
• Experience in developing high-performance kernels for CPUs
• Experience in developing ML or traditional compilers
• Experience with TPU
• Strong publication records in top HPC/ML/DL or computer system and architecture venues

Benefits

• An open and inclusive culture and work environment
• Fully stocked kitchen at the office
• Full health and dental benefits
• Parental Leave top-up for 6 months
• Continuous education budget
• Generous vacation - we're not saying unlimited, but if you need extra time to recharge, just ask