Job Title: Senior Python Developer – AI/ML Model SDKs
Location: Alpharetta, GA [Locals Only - F2F Interviews]
Duration: 12+ Months
Experience Required: 7+ Years
Role Overview
We are seeking a Senior Python Developer to design, build, and release high quality Python Model SDKs used by data scientists and machine learning engineers. This role focuses on developing scalable and developer friendly libraries that support the full AI and machine learning lifecycle including model training, experimentation, validation, and deployment.
The ideal candidate will have deep expertise in Python, strong software engineering fundamentals, and experience working with AI and machine learning workflows, ensuring SDKs are production ready and optimized for real world usage.
Key Responsibilities
Design, develop, and maintain Python SDKs that simplify AI and machine learning model training, evaluation, and deployment workflows.
Build SDKs optimized for notebook based development environments such as Jupyter, Google Colab, and Visual Studio Code.
Develop clean, modular, and extensible APIs that support multiple model types and machine learning frameworks.
Package and release SDKs following best practices in versioning, dependency management, and backward compatibility.
Ensure SDKs are production ready and support deployment, inference, monitoring, and configuration management.
Collaborate with data scientists, machine learning engineers, and MLOps teams to translate requirements into SDK features.
Develop and maintain comprehensive unit, integration, and contract tests using modern testing frameworks.
Create and maintain developer documentation, usage examples, and notebooks.
Enforce software engineering best practices including code reviews, CI CD pipelines, linting, and performance optimization.
Manage the complete SDK lifecycle from design and development through release and ongoing maintenance.
Required Skills and Qualifications
Education
Bachelor’s degree in Computer Science, Information Technology, or a related field is preferred.
Experience
7+ years of professional experience in Python development.
Experience building, maintaining, and distributing Python libraries or SDKs for developers.
Strong understanding of AI and machine learning workflows including training, validation, inference, and deployment.
Technical Skills
Strong expertise in Python programming and software engineering practices.
Experience designing APIs optimized for interactive notebook environments.
Experience with Python packaging tools such as setuptools, poetry, or pip.
Experience with semantic versioning and package distribution via PyPI or internal repositories.
Strong understanding of software design principles including SOLID and clean architecture.
Experience with testing frameworks such as pytest.
Familiarity with CI CD pipelines and automated release processes.
Experience working with machine learning frameworks such as TensorFlow, PyTorch, Scikit learn, or XGBoost.
Experience with MLOps practices including model versioning, feature stores, model registries, and monitoring.
Experience deploying machine learning models on cloud platforms such as AWS, Google Cloud Platform, or Microsoft Azure.
Familiarity with containerization and orchestration tools such as Docker and Kubernetes.
Soft Skills
Strong communication and collaboration skills.
Ability to work effectively with cross functional teams including data scientists and MLOps engineers.
Strong problem solving and analytical thinking abilities.
Attention to detail and commitment to high quality software delivery.
Preferred Qualifications
Experience designing developer focused APIs and SDK usability patterns.
Contributions to open source projects or publicly released Python packages.
Experience working in large scale AI and machine learning platform environments.
Never miss an opportunity! Create an alert based on the job you applied for.
