Senior Python Developer – AI/ML SDK

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 / GCP / 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