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. Role & responsibilities Architect, train, fine-tune, and distill 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