AI/ML Engineer - Python Programming

Job Description :

We are seeking a skilled AI/ML engineer in Pune with 6+ years of experience in designing and developing scalable enterprise applications.

About Us :

CodeVyasa is a mid-sized product engineering company that works with top-tier product and solutions organizations such as McKinsey, Walmart, RazorPay, Swiggy, and others. We are a team of 550+ engineers, driving innovation across Product & Data Engineering, focusing on Agentic AI, RPA, Full Stack, and GenAI-based solutions.

Key Responsibilities :

- Design, develop, and deploy scalable machine learning and deep learning models for production environments.

- Build end-to-end AI/ML pipelines, from data preprocessing and model training to deployment and monitoring.

- Develop RESTful APIs using FastAPI for serving ML models.

- Work closely with data engineers, software developers, and business stakeholders to deliver AI-driven solutions.

- Optimize machine learning models for performance, scalability, and reliability.

- Implement CI/CD pipelines and MLOps best practices for model deployment and lifecycle management.

- Process and analyze large-scale datasets using distributed computing frameworks.

- Evaluate, fine-tune, and deploy models for Computer Vision, NLP, or Predictive Analytics use cases.

- Ensure high-quality code, documentation, and adherence to engineering best practices.

Tech Stack & Qualifications :

- Strong proficiency in Python programming.

- Hands-on experience with ML/DL libraries: PyTorch, TensorFlow, OpenCV, Pandas, NumPy.

- Experience in at least two domains : Computer Vision, Natural Language Processing (NLP), Predictive Modeling.

- Strong experience developing APIs using FastAPI and REST APIs.

- Hands-on experience with Docker and containerized deployments.

- Experience with Big Data technologies such as PySpark, Hadoop, Hive, or Dask.

- Strong understanding of SQL and NoSQL databases.

Why Join CodeVyasa?

- Work on innovative, high-impact projects with leading global clients.

- Exposure to modern technologies, scalable systems, and cloud-native architectures.

- Continuous learning and upskilling opportunities through internal and external programs.

- Supportive and collaborative work culture with flexible policies.

- Competitive salary and comprehensive benefits package.

- Free healthcare coverage for employees and dependents.