AI/Machine Learning Engineer
apple
Job Description
- Bachelor’s Degree or Equivalent with 4+ years of experience in AI/ML & Data engineering
- Ability to design systems that go from raw data → structured insight → autonomous action, closing the loop between data engineering and AI decision-making
- Design and implement RAG (Retrieval-Augmented Generation) pipelines that feed structured data, user personas, and domain-specific context into LLMs for accurate, grounded outputs
- Deep understanding of embeddings, vector databases, and semantic search to power retrieval layers in AI applications
- Hands-on experience building AI agents with memory, planning, tool use, and execution loops using frameworks such as LangChain, LlamaIndex.
- Proven expertise with transformer-based models (BERT, GPT, LLaMA, etc.) including fine-tuning, prompt engineering, and understanding of their underlying attention mechanism
- Proficiency with ML frameworks such as PyTorch or TensorFlow, including model training, optimization, and serving pipelines
- Experience working with data at scale (peta bytes) with big data tech stack and advanced programming languages e:g Python, Scala.
- Database development experience with Relational or MPP/distributed systems such as Snowflake, SingleStore along with expertise in SQL and Advance SQL.
- Experience in designing and building dimensional data models to improve accessibility, efficiency and quality of data
Preferred Qualifications
- Experience building multi-agent AI systems and agentic workflows that coordinate across data retrieval, reasoning, and action steps autonomously is a plus
- Experience in modern cloud warehouse, data lakes and implementation experience on any of the cloud platforms like AWS/GCP/Azure - preferably AWS.
- Have continuous focus to Brainstorm and Design various POCs using AI/ML Services for new or existing enterprise problems