Senior Engineer, Machine Learning Application Developer

What you'll do

• Develop neural rendering applications and machine learning software for mobile GPUs.
• Implement ML operator kernels including GEMM, convolution, and activations using Vulkan, OpenGL, and OpenCL.
• Optimize performance, efficiency, and resource utilization for AI workloads on GPU hardware.
• Analyze software performance and hardware resource usage to identify and resolve bottlenecks.
• Perform low-level performance analysis and assembly-level investigation to improve execution efficiency.
• Translate hardware constraints into optimized software solutions through collaboration with architecture teams.
• Develop API-level software to bridge machine learning workloads and GPU hardware capabilities.

What we're looking for

• Bachelor's degree in Computer Science or Computer Engineering with 3+ years of experience is required.
• Master's degree or Ph.D. with 2+ years of experience is accepted as an equivalent qualification.
• Strong programming skills in C, C++, and Python are required.
• Proficiency in Vulkan, OpenGL, OpenCL, PyTorch, and TensorFlow is required.
• Experience developing neural rendering applications at the API level is required.
• Knowledge of machine learning operators such as GEMM, convolution, and activations is required.
• Experience with GPU hardware architecture and low-level performance profiling/analysis is required.
• Applicants must be authorized to access information subject to U.S. export control laws.