Senior Software Engineer (Deep Learning)

Job Title: Senior Software Engineer (Deep Learning)

Location: Onsite in Pittsburgh or Burlingame is Heavily Preferred | Open to Remote US

Duration: 6 months contract

Pay range: $110 - $115/hr on W2

We are looking for a seasoned Senior Software Engineer (SWE V) with a research engineering skillset to spearhead code improvement initiatives. The ideal candidate should possess a robust background in software development, particularly in Python, Deep Learning (DL) theoretical knowledge, experience training big DL models, and code optimization. This includes continuous integration and testing, experience with GPU coding, PyTorch, and backend ML systems.

Our team focuses on transforming Deep Learning 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

• Codebase migrations
• Enhancing code quality
• Increasing test coverage through unit and integration testing
• Code refactoring
• Optimizing backend workflow orchestration
• Additional engineering tasks such as setting up dashboards and alerts, and assisting with on-call workloads
• Interact with scalable and distributed training algorithms and efficient data loading for large scale deep learning
• Collaborate with both research and maturation teams to push research to products
• Most of the tasks detailed would be performed on DL pipelines

Must-have:

• 6+ years of experience in Python, especially with large codebases, preferably in Big Tech or mid-to-large companies
• 3+ years’ on PyTorch and Deep Learning
• Experience in training big deep learning models with PyTorch
• Proficiency in continuous integration and testing
• Knowledge of code optimization techniques
• Familiarity with GPU coding and backend ML systems (a plus)
• Strong problem-solving skills and attention to detail, with the ability to independently overcome technical roadblocks
• Experience working collaboratively and communicate effectively across functional teams
• Capability to work independently and within a team

Nice to have:

• Experience in maturing machine learning systems, such as defining evaluation metrics, building evaluation systems, scaling up pipelines and tooling
• Experience in creating and maintaining public projects (e.g., GitHub)
• Code Optimization (CPU and GPU), GPU Coding

Education:

Master's degree or higher in computer vision, machine learning, deep learning, or a related field

Recruitment Process:

• Screening Interview and Python Assessment
• 2 - 45 mins. coding interview with the hiring manager
• Offer
• Onboarding