Ai Ml Engineer

Job description

We're seeking an experienced AI/ML Engineer who thrives on curiosity and innovation to join our dynamic engineering team. This role offers an exciting chance to tackle impactful real-world machine learning challenges and play a key part in developing advanced, state-of-the-art AI solutions.

Experience in developing Generative AI models

Role & responsibilities :

• Architect, train, fine-tune, and still LLMs and deep-learning models for NLP, generative AI, and agentic AI applications.
• Assist in designing, developing, and training machine learning models using structured and unstructured data
• Conduct thorough testing, evaluation, benchmarking, and validation to ensure performance and robustness
• Deploy models into production using MLOps pipelines (CI/CD, versioning, monitoring)
• Optimize models and inference pipelines for latency, throughput, and compute efficiency
• Mentor junior engineers and collaborate cross-functionally with data engineers, DevOps, product managers
• Curate, cleanse, and transform extensive datasets to prepare them for robust model development.
• Conduct insightful exploratory data analysis and apply advanced statistical modeling techniques.
• Execute systematic experiments, optimize hyperparameters, and rigorously assess models using established performance metrics.
• Maintain comprehensive documentation detailing model architectures, data pipelines, and experimental outcomes.
• Develop and integrate AI-enhanced tools including web-search functionalities, intelligent image editing, and related applications.
• Support in integrating AI/ML models into production environments

Skill Sets We Require-

• Frameworks: PyTorch (latest, including Torch Dynamo), TensorFlow 2.x
• Finetuning & Distillation: Implement teacher student training, supervised finetuning, response based distillation and Transorfermers.
• Model Lifecycle
• Hands-on experience with Python and ML libraries such as scikit-learn, pandas, NumPy.
• Understanding of probability, statistics, and linear algebra
• APIs & Storage: RESTful APIs, ONNX, SQL/NoSQL databases, S3 Buckets.
• CI/CD: Git, GitHub Actions, Docker Registry, automated testing & deployment (Good to have)

Required Qualifications-

• BTech/BE in Computer Science or Equivalent Field
• 4+ years of hands-on experience in training, tuning, and deploying LLMs and deep-learning models.
• Proficient with PyTorch, Transformer frameworks