Senior Python Developer - AI/ML SDK & Platform Engineering

Location: Alpharetta, GA

Job Summary

We are looking for a highly skilled Senior Python Developer to design and build AI/ML Model SDKs that enable seamless model development, training, deployment, and monitoring.

The ideal candidate will have strong expertise in Python development, SDK/library design, and AI/ML workflows, along with experience building developer-friendly tools used by data scientists and ML engineers.

Key Responsibilities

SDK & Platform Development

• Design, develop, and maintain Python SDKs/libraries for:
  • Model training
  • Experimentation
  • Validation
  • Deployment
• Build developer-friendly APIs supporting multiple ML frameworks

AI/ML Workflow Enablement

• Support full ML lifecycle:
  • Training
  • Evaluation
  • Inference
  • Deployment
• Optimize SDKs for notebook environments:
  • Jupyter
  • Google Colab
  • VS Code

Architecture & Engineering

• Implement clean, modular, and scalable APIs
• Follow best practices:
  • SOLID principles
  • Clean architecture
• Ensure performance, reliability, and scalability

Packaging & Release Management

• Manage SDK lifecycle:
  • Versioning (Semantic versioning)
  • Dependency management
  • Backward compatibility
• Publish packages using:
  • PyPI / internal registries

Testing & Quality

• Write:
  • Unit tests
  • Integration tests
  • Contract tests (pytest)
• Enforce:
  • Code reviews
  • Linting
  • CI/CD best practices

Collaboration

• Work closely with:
  • Data Scientists
  • ML Engineers
  • MLOps teams
• Translate requirements into SDK features
• Create documentation, examples, and notebooks

Required Skills

Core Skills (Must Have)

• Strong Python development (7 years)
• Experience building Python libraries / SDKs
• API design for developer tools

AI/ML & Data

• Understanding of:
  • Model training
  • Validation
  • Inference
  • Deployment
• Experience with:
  • TensorFlow / PyTorch / Scikit-learn / XGBoost

Packaging & Tools

• setuptools / poetry / pip
• Semantic versioning
• PyPI or internal package distribution

Testing & DevOps

• pytest
• CI/CD pipelines
• Git-based workflows

Cloud & MLOps

• AWS / Google Cloud Platform / Azure
• Model deployment & monitoring
• Feature stores / model registries

Containers

• Docker
• Kubernetes

Preferred Skills

• Open-source contributions / published Python packages
• Experience with developer SDK usability patterns
• Strong documentation and communication skills

Experience Required

• 7 years of Python development experience
• Proven experience building SDKs or reusable libraries