We’re hiring: AI / ML Engineer | 4–5 Years Experience

We are building advanced AI technology for next generation design and manufacturing applications.

We are looking for a robust AI / ML Engineer with 4–5 years of hands-on experience who can take complex AI research and turn it into scalable, production ready systems.

You will work across computer vision, generative AI, deep learning and 3D applications, with a robust focus on research, experimentation and production implementation.

What we’re looking for
Strong Python and PyTorch experience
Transformers and attention based architectures
CNNs and Vision Transformers
Generative AI and diffusion models
Computer vision and image processing
Experience with 3D applications and geometry based workflows
Model training, fine tuning and evaluation
Multi GPU / distributed training and CUDA optimisation
Hugging Face Transformers / Diffusers
MLflow for experiment tracking, model versioning and model lifecycle management
Apache Airflow for workflow and pipeline orchestration
Building and maintaining end to end ML pipelines
Data preprocessing, training,



evaluation and inference pipelines
Model registry and reproducible experimentation
GPU inference and memory optimisation
Docker, Linux and Git
CI/CD and MLOps practices for machine learning systems
Cloud and GPU infrastructure experience
Model monitoring, logging and performance tracking
Ability to understand research papers and implement recent architectures independently

Bonus experience
PyTorch3D, Open3D, Trimesh or similar frameworks
Blender Python or 3D automation
CAD / BREP / NURBS concepts
Differentiable rendering
Kubernetes and container orchestration
AWS, GCP or Azure ML infrastructure
Experience taking research models from prototype to production

We are looking for someone who is technically strong, research driven and equally comfortable with model development and production ML engineering.

Experience: 4–5 years

Domain: AI / ML, Computer Vision, Generative AI, 3D, MLOps

Location: Mumbai
Hiring #MachineLearning

📌 Machine Learning Engineer Mumbai
🏢 Lumex Group
📍 Mumbai