Machine Learning Engineer

Python Software Engineer

📍 Location: Remote – Burlingame, CA & Pittsburgh, PA

⏳ Duration: 8+ months (Possibility of extension)

💰 Pay Rate: $110-$115/hr on W2

Job Overview:

We are seeking a Python Software Engineer with extensive experience working on large-scale codebases. The ideal candidate will be responsible for codebase migrations, enhancing code quality, increasing test coverage, and optimizing backend workflow orchestration. This role will involve close collaboration with research and maturation teams to integrate cutting-edge research into production-ready solutions.

Key Responsibilities:

✔️ Codebase migrations and refactoring

✔️ Enhancing code quality and increasing test coverage (unit & integration testing)

✔️ Optimizing backend workflow orchestration

✔️ Setting up dashboards, alerts, and assisting with on-call workloads

✔️ Working with scalable and distributed training algorithms

✔️ Efficient data loading for large-scale deep learning models

✔️ Collaborating with research and engineering teams to push research into production

Must-Have Skills:

✅ 6+ years of Python experience, particularly with large codebases (Big Tech/mid-to-large companies preferred)

✅ Proficiency in continuous integration and testing

✅ Strong code optimization techniques

✅ Familiarity with GPU coding, PyTorch, and backend ML systems (preferred)

✅ Strong problem-solving skills and ability to overcome technical roadblocks

✅ Excellent collaboration and communication skills across teams

✅ Ability to work independently and in a team environment

Nice-to-Have Skills:

➕ Experience in training deep learning models

➕ Experience in machine learning systems (evaluation metrics, scaling pipelines, tooling)

➕ Experience with open-source projects (GitHub contributions)

➕ Expertise in Code Optimization/GPU Coding

➕ Hands-on PyTorch experience

Interview Process:

📌 Rounds: 3

1️⃣ Screening Interview

2️⃣ Two (45-min) Coding Interviews

🚀 Ready to take on a high-impact role in a dynamic team? Apply today! 🚀