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 strong 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 new 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