Ai Ml Engineer

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, andstill 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 TorchDynamo), TensorFlow 2.x

• Finetuning & Distillation:

  Implement teacherstudent training, supervised finetuning, responsebased 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