Overview:Senior Python Developer - AI/ML SDK & Platform EngineeringLocation: Alpharetta, GA
Job SummaryWe 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 / GCP / Azure
- Model deployment & monitoring
- Feature stores / model registries
Containers 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