Careers

AI/MLEngineer

Years of Experience: 
4-10 Years
Skills Stack: 
Python, Pandas, PySpark, NumPy, TensorFlow, PyTorch, MLflow
Shift Timings: 
05:30 am - 05:30 am ()
Role and Responsibility Details: 
Job Description: 

Key Responsibilities

  • Design, develop, and deploy ML models for predictive analytics, computer vision, or NLP use cases.
  • Build scalable data ingestion and preprocessing pipelines using Python (Pandas, PySpark, etc.).
  • Implement and optimize ML workflows using frameworks like TensorFlow, PyTorch, or scikit-learn.
  • Collaborate with data engineers and data scientists to integrate models into production systems.
  • Automate model retraining, monitoring, and performance tracking.


Required Skills & Qualifications

  • Bachelor's or Master's degree in Computer Science, Data Science, Engineering, or a related field.
  • 4-9 years of hands-on experience in Python-based AI/ML development.
  • Good to have understanding of machine learning algorithms, model evaluation, and feature engineering.
  • Proficiency with Python libraries such as NumPy, Pandas, scikit-learn, TensorFlow, PyTorch, and MLflow.
  • Experience with data processing frameworks such as PySpark or Dask.
  • Familiarity with cloud platforms (AWS, Azure, GCP) and containerization tools (Docker, Kubernetes).
  • Experience with API integration and MLOps tools (Databricks, Airflow, SageMaker, etc.).
  • Understanding of version control (Git) and CI/CD pipelines
Qualifications: 
B.E
Locations: 
Pune
Contact: 
Recruitment Team
Email: tdg-recruitment@thedigitalgroup.com