Software Engineer, ML Dev Enablement
New
M
MotionalAutonomous Driving
Las Vegas, Nevada, United States; Remote U.S.Full-TimeMiddle
Salary$123,000 — $163,500 USD
Apply NowOpens the employer's application page
Job Details
- Required Skills
- AWSPythonKubernetesPyTorchC++GoLLMMLOpsDistributed Systems
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 with building on AWS services or other cloud platforms.
- Experience with container orchestration using Kubernetes.
- Experience with the various stages of the ML development lifecycle.
- Hands-on experience with ML model profiling and performance optimization (bonus).
- Experience managing or working with high-performance compute resources/GPUs (bonus).
- Experience with ML frameworks such as PyTorch or Ray (bonus).
- Experience building, integrating, or enhancing Agentic AI systems and LLM-driven developer tools (bonus).
Responsibilities
- Design, develop, and enhance Agentic AI tools and systems to automate workflows, streamline the ML lifecycle, and empower developer productivity.
- Drive the continuous development of our core ML infrastructure and existing CDE platform, leveraging Kubernetes to build robust, high-scale distributed solutions.
- Partner closely with ML Researchers to profile and optimize distributed training jobs (PyTorch/DDP) and data pipelines.
- Resolve system-level bottlenecks such as data loading (I/O), memory management, and network communication overhead.
- Maximize GPU utilization and training throughput.
- Partner with ML engineers and data scientists to bridge the gap between underlying infrastructure and model development.
View Full Description & ApplyYou'll be redirected to the employer's site