ML Solutions Engineer

TensorWave(2 months ago)

RemoteFull TimeSenior$151,822 - $204,510 (estimated)Engineering

About this role

The ML Solutions Engineer at TensorWave focuses on GPU portability and performance optimization, specifically migrating and enhancing CUDA workloads to run efficiently on ROCm and AMD hardware. This senior role involves collaborating with internal teams and third-party developers, profiling and debugging GPU kernels, and contributing to ROCm enablement across open-source ML frameworks. The engineer provides technical guidance on best practices, creates documentation, and acts as a liaison to ensure customer needs are met effectively while advancing the ROCm ecosystem.

View Original Listing

Required Skills

CUDA
ROCm
GPU Programming
Performance Optimization
Kernel Development
GPU Profiling
Distributed ML Workloads
Technical Guidance
Documentation Creation
Customer Liaison

+8 more

About TensorWave

tensorwave.com

TensorWave is a cloud computing platform specializing in artificial intelligence (AI) and high-performance computing (HPC) services powered by AMD Instinct™ GPUs. The company provides a scalable and memory-optimized infrastructure designed to facilitate the deployment and management of demanding AI workloads, including low-latency inference and large language models. With a focus on efficiency and cost-effectiveness, TensorWave's offerings include bare-metal solutions and managed inference services, tailored to meet the needs of enterprises looking to harness the power of next-generation AI technologies.