Senior Python Developer AI/ML SDK

Dice is the leading career destination for tech experts at every stage of their careers. Our client, Purple Drive Technologies LLC, is seeking the following. Apply via Dice today!

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