Senior ML Engineer

Position: Senior ML Engineer at Waabi
Join Waabi as a Senior ML Engineer, specializing in optimization for self-driving technologies. Utilize your skills in deep learning and coding to drive innovation in AI systems.

This role involves collaborating with a world-class team to enhance the self-driving capability of Waabi’s systems. You will build distributed training frameworks and analyze model performance to ensure stability and efficiency. Your expertise in identifying and integrating cutting-edge technologies will be vital for advancing our AI-driven solutions.

Key Responsibilities:

• Work closely with autonomy engineers on system scalability

• Create frameworks for research and production environments

• Analyze runtime and memory for performance improvements

• Discover technologies to enhance training frameworks

• Implement CUDA kernels where necessary

Requirements:

• MS/PhD or Bachelor's degree and 6+ years experience

• Strong coding skills in Python, C++, or Rust

• Experience with PyTorch deep learning framework

• Knowledge across data processing and model deployment stages

• Proficient in profiling tools like PyTorch Profiler

Shape the future of self-driving technology with your expertise at Waabi.
#J-18808-Ljbffr