Python ML Developer # VR/AR, gaming, or avatar systems #onsite #W2

Role: Python ML Developer

Location: Onsite in Burlingame, CA or Pittsburgh, PA

Duration: Long Term

Candidate should have experience on these and resumes to speak about all these:

Must-Have Skills (in simple terms):

• 6 years of Python development (clean, testable code)
• Strong ML infrastructure support experience (model pipelines, jobs, infra tools)
• Worked with ML researchers or R&D teams before
• Experience with GPU pipelines, debugging, automation
• Familiarity with VR/AR, gaming, or avatar systems
• Hands-on with Docker, Kubernetes, CI/CD, or other infra tooling
• Exposure to GPU training, job scheduling, and performance optimization

Target Companies (for sourcing):

• Meta, Google, Amazon, Apple, Nvidia, Microsoft
• Startups or mid-sized companies in ML Infra, AI R&D, VR/AR
• ML tooling or avatar/telepresence startups

Job Overview

We are looking for a seasoned Senior Software Engineer (SWE V) to spearhead code improvement initiatives. The ideal candidate should possess a robust background in software development, particularly in Python, continuous integration and testing, and code optimization. Experience with GPU coding, PyTorch, and backend ML systems is advantageous. Our team focuses on transforming research codebases into product-ready states. The successful candidate will primarily code in Python. While experience with backend systems and managing large codebases is beneficial, it is not mandatory.

Key Responsibilities

1. Codebase migrations

2. Enhancing code quality

3. Increasing test coverage through unit and integration testing

4. Code refactoring

5. Optimizing backend workflow orchestration

6. Additional engineering tasks such as setting up dashboards and alerts, and

assisting with on-call workloads

7. Interact with scalable and distributed training algorithms and efficient data

loading for large scale deep learning

8. Collaborate with both research and maturation teams to push research to

products

Must-Have Skills

1. 6 years of experience in Python, especially with large codebases, preferably in Big Tech or mid-to-large companies

2. Proficiency in continuous integration and testing

3. Knowledge of code optimization techniques

4. Familiarity with GPU coding, PyTorch, and backend ML systems (a plus)

5. Strong problem-solving skills and attention to detail, with the ability to independently overcome technical roadblocks

6. Experience working collaboratively and communicate effectively across functional teams

7. Capability to work independently and within a team

Nice-to-Have Skills

1. Experience in training deep learning models is a plus

2. Experience in machine learning systems, such as defining evaluation metrics, building evaluation systems, scaling up pipelines and tooling

3. Experience in creating and maintaining public projects (e.g., GitHub)

4. Code Optimization/GPU Coding

5. PyTorch Experience

Degrees/Certifications Bachelor s in software engineering preferred

Day-to-Day Responsibilities

Work includes

1. Codebase migrations

2. Improving code quality

3. Increase test coverage with both unit and integration testing

4. Code refactoring

5. Improve how backend workflows are orchestrated

6. Other better engineering work like setting up dashboards and alerts, helping with oncall workloads etc