Ai/ml engineer

Job Description

We are seeking a skilled Ai/ml engineer| pune having 6+ yrs with strong 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.

Required Skills & Qualifications

• Strong proficiency in Python programming.
• Hands-on experience with ML/DL libraries including:
• PyTorch
• TensorFlow
• OpenCV
• Pandas
• NumPy
• Experience in at least two of the following 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.