Start Your Search Here

Job Search

Baseten

Montreal (Administrative Region) / Global

High-Performance GPU Engineer at Baseten

Job Description

Become a High-Performance GPU Engineer at Baseten, where your skills drive the optimization of AI workloads. Help create next-level solutions and influence technology industry-wide.

Baseten is searching for a talented engineer to design and implement GPU kernels that enhance machine learning operations. You’ll utilize CUDA and advanced performance methods to optimize systems serving millions of users. This role suits engineers passionate about low-level details and eager to work on groundbreaking AI technologies.

Key Responsibilities:

• Design and implement GPU kernels for ML tasks

• Optimize applications with CUDA and advanced programming techniques

• Explore features like compute/communication overlap and quantization

• Mitigate performance bottlenecks using specific tools

• Contribute to internal and open-source GPU libraries

Requirements:

• In-depth understanding of GPU architecture mechanisms

• Proficient in C++ and GPU profiling methodologies

• Knowledge of memory hierarchy and optimization strategies

• Familiar with relevant GPU libraries and frameworks

• Research experience on GPU performance preferred

Make your mark on AI technology as you optimize GPU performance at Baseten.

#J-18808-Ljbffr

Apply Now

Similar Opportunities

View all jobs