Software Development Engineer, ML Acceleration, Trainium AI Systems, Annapurna Labs

This job was posted over 30 days ago, so it has most likely closed. See newer jobs like it.

About this kind of role

What a Software Development Engineer, ML Acceleration, Trainium AI Systems, Annapurna Labs usually does. The employer's own details are in the listing.

Typically does: This role focuses on optimizing machine learning workloads for specialized hardware, often involving performance analysis, code profiling, and algorithm tuning. Software Development Engineers in this area work to improve the efficiency and speed of training and inference processes. They may also be involved in developing tools and libraries to support hardware acceleration. Collaboration with hardware engineers and machine learning researchers is common.

Tools and skills: Proficiency in programming languages like C++, Python, and CUDA, along with experience in performance profiling tools, machine learning frameworks (e.g., TensorFlow, PyTorch), and hardware-specific optimization techniques is generally expected.

Good fit for: Individuals with a strong background in computer science, a passion for machine learning, and a desire to improve the performance of complex systems typically thrive in this role.