Backend Software Engineer: Python-PyTorch

Bayside Solutions; United States; On-site; USD 114,400 - 135,200; Full time

Job summary

A leading solutions provider is seeking a Backend Software Engineer specializing in Python and PyTorch to join their central ML/AI platform team. This role involves designing and supporting scalable ML infrastructure for various organizational needs. Ideal candidates should possess strong Python experience, hands-on proficiency with PyTorch, and a solid understanding of machine learning fundamentals. This is a W2 contract position with a competitive salary range of $114,400 to $135,200 per year.

Qualifications

• Strong experience with Python, including writing production-quality, maintainable code.
• Hands-on experience with PyTorch in real-world ML systems (training and/or inference).
• Solid understanding of ML fundamentals, including embeddings and representation learning.

Responsibilities

• Design, develop, and enhance ML infrastructure services supporting training and inference.
• Implement small to medium-sized features across existing ML platform components.
• Communicate technical decisions and trade-offs clearly.

Skills

Python

PyTorch

Machine Learning Infrastructure

Debugging

Cross-Functional Collaboration

Job description

Backend Software Engineer: Python-PyTorch

W2 Contract

Salary Range: $114,400 - $135,200 per year

Job Summary

We are seeking experienced AI / ML Infrastructure Engineers to join a central ML/AI platform team responsible for building foundational services used by multiple internal and external product organizations. This role focuses on designing, extending, and supporting scalable ML infrastructure that powers both traditional machine learning and modern LLM-based workflows.

Duties and Responsibilities

• Design, develop, and enhance ML infrastructure services supporting training, inference, experimentation, and embeddings lifecycle management.
• Implement small to medium-sized features across existing ML platform components.
• Take customer use cases end-to-end, including investigation, debugging, and resolution of issues.
• Work across the ML stack, collaborating closely with applied scientists and downstream product teams.
• Diagnose and resolve issues in production ML systems.
• Navigate ambiguous requirements and proactively unblock work by seeking context or collaboration when needed.
• Clearly communicate technical decisions, trade-offs, and implementation rationale.

Requirements and Qualifications

• Strong experience with Python, including writing production-quality, maintainable code
• Hands-on experience with PyTorch in real-world ML systems (training and/or inference)
• Solid understanding of ML fundamentals, including:
• Embeddings and representation learning
• Experimentation and evaluation workflows
• Experience debugging and maintaining complex, distributed systems
• Ability to reason through problems, explain solutions, and articulate trade-offs
• Comfort operating in environments with ambiguity and incomplete requirements

Preferred Qualifications

• Experience building or supporting ML infrastructure platforms
• Familiarity with feature stores, experimentation frameworks, or inference services
• Exposure to large-scale, multi-team ML environments
• Prior work supporting both research and production ML use cases

Desired Skills and Experience

Python, PyTorch, Machine Learning Infrastructure, AI/ML Platforms, Model Training, Model Inference, LLM Workflows, Embeddings, Representation Learning, Experimentation Frameworks, ML Evaluation, Inference Services, Feature Stores, Distributed Systems, Scalable Systems Design, Production ML Systems, Debugging and Troubleshooting, ML Platform Development, Infrastructure Services, End-to-End Problem Solving, Cross-Functional Collaboration, Applied Science Collaboration, Research-to-Production ML, Ambiguous Requirements Handling, Technical Communication, Trade-off Analysis, Maintainable Production Code

Bayside Solutions, Inc. is not able to sponsor any candidates at this time. Additionally, candidates for this position must qualify as a W2 candidate.

Bayside Solutions, Inc. may collect your personal information during the position application process. Please reference Bayside Solutions, Inc. s CCPA Privacy Policy at www.baysidesolutions.com.