Software Engineer - AI/ML

About the Job

As an AI/ML Engineer, you will be responsible for designing, validating, and integrating cutting-edge machine learning models and algorithms. Collaborate closely with cross-functional teams, including data scientists, to recognize and establish project objectives. Oversee data infrastructure maintenance, ensuring streamlined and scalable data operations. Stay updated with advancements in AI and propose their integration for operational enhancement. Effectively convey detailed data insights to non-technical stakeholders. Uphold stringent data privacy and security protocols. Engage in the full lifecycle of AI projects, spanning from ideation through deployment and continuous upkeep.

Core Responsibilities

• Develop, validate, and implement machine learning models and algorithms.
• Collaborate with data scientists and other stakeholders to understand and define project goals.
• Maintain data infrastructure and ensure scalability and efficiency of data-related operations.
• Stay abreast of the latest developments in the field of AI/Client and recommend ways to implement them in our operations.
• Communicate complex data findings clearly and understandably to non-technical stakeholders.
• Adhere to data privacy and security guidelines.
• Participate in the entire AI project lifecycle, from concept to deployment and maintenance.

Required Skills

• Strong grasp of computer architecture, data structures, system software, and machine learning fundamentals.
• Experience with mapping NLP models (BERT and GPT) to accelerators and awareness of trade-offs across memory, bandwidth, and compute.
• Experience with Vector databases like Qdrant, ChromaDB, Pinecone, PGVector or similar
• Experience with Large language Models like GPT 3.5, GPT 4, GPT 4.o, Llama, Gemini, Mistral etc
• Experience in design, build, and optimization of agentic AI workflows that enable autonomous reasoning, decision-making, and tool interaction using frameworks like LangChain, LangGraph, CrewAI, etc.
• Experience in implementation and fine-tuning Generative AI solutions (e.g., LLMs, multimodal models) for various business use cases.
• Experience in LLM integration framework like Llamaindex, Ragas, etc
• Experience with ML Models from definition to deployment, including training, quantization, sparsity, model preprocessing, and deployment.
• Proficiency in Python development in a Linux environment and using standard development tools.
• Experience with deep learning frameworks (such as PyTorch, Tensorflow, Keras, Spark).
• Working knowledge of Artificial Neural Networks (ANNs), Convolutional Neural Networks (CNNs), Recurrent Neural Networks (RNNs), and Generative Adversarial Networks (GANs).
• Experience in training, tuning, and deploying ML models for Computer Vision (e.g., ResNet), and/or Recommendation Systems (e.g., DLRM).
• Experience deploying ML workloads on distributed systems.
• Self-motivated team player with a strong sense of ownership and leadership.
• Strong verbal, written, and organizational skills for effective communication and documentation.
• Research background with a publication record.
• Work experience at a cloud provider or AI compute/sub-system company.
• Knowledge of cloud computing platforms and services, such as AWS, Azure, or Google Cloud.
• Experience with information security and secure development best practices.

Qualifications

• Bachelor's or higher degree in Computer Science, Engineering, Mathematics, Statistics, Physics, or a related field.
• 2-4 yrs of hands-on experience in AI/ML