onsite
AI Engineer - Petals Automation
AI Engineer
AI Engineer responsible for building and deploying machine learning models using Python, TensorFlow, and PyTorch. Designs end‑to‑end pipelines, containerizes services with Docker, orchestrates with Kubernetes, and collaborates with cross‑functional teams to deliver AI solutions.
About the role
Key Responsibilities
- Design, develop, and optimize machine learning models for classification, regression, and clustering tasks.
- Implement end‑to‑end ML pipelines using Python, scikit‑learn, TensorFlow, and PyTorch.
- Containerize models and services with Docker, and deploy them on Kubernetes clusters.
- Expose model functionality via RESTful APIs using Flask or FastAPI.
- Collaborate with data engineers and product teams to integrate AI solutions into production workflows.
Requirements
- 1–2 years of hands‑on AI/ML development experience.
- Strong programming skills in Python and familiarity with NumPy and Pandas.
- Proficiency with ML frameworks: scikit‑learn, TensorFlow, and PyTorch.
- Experience with version control (Git) and CI/CD pipelines.
- Knowledge of containerization (Docker) and orchestration (Kubernetes) for model deployment.
Skills
pythontensorflowpytorchdockerkubernetes