Software Engineer, ML Dev Enablement

Software Engineer, ML Dev Enablement

Location: Las Vegas
Workplace: Remote
Employment: Full Time
Salary: USD 123,000–163,500/yr

Apply on the employer’s site

Posted 1mo ago

The employer’s full description could not be read from their board. This is a summary of the posting — follow the apply link for the original.

Responsibilities

Design, develop, and enhance Agentic AI tools and systems to automate workflows
Drive continuous development of core ML infrastructure and CDE platform using Kubernetes
Profile and optimize distributed training jobs (PyTorch/DDP) and data pipelines
Resolve system-level bottlenecks such as data loading, memory management, and network communication
Partner with ML engineers and data scientists to bridge infrastructure and model development

Requirements

BS or MS in Computer Science or related field
Strong knowledge of software engineering principles and distributed systems
Strong proficiency with Python, Go, or C++
Experience building on AWS services or other Cloud platforms
Experience with container orchestration using Kubernetes
Experience with various stages of the ML development lifecycle

Preferred

Hands-on experience with ML model profiling and performance optimization for distributed training
Experience managing or working with high-performance compute resources (GPUs)
Experience with ML frameworks such as PyTorch or Ray
Experience building, integrating, or enhancing Agentic AI systems and LLM-driven developer tools

Skills

Python
Go
C++
AWS
Kubernetes
PyTorch
Ray
DDP
LLM