AI/ML Engineer: LLMs, Agentic AI & RAG Expert

Tech Aalto Pte ltd; Singapore; On-site; SGD 120,000 - 190,000; Full time

Job summary

Tech Aalto Pte ltd seeks an experienced AI/ML Engineer to design, implement, and optimize prompt strategies for cutting-edge large language models. You will build RAG datasets, fine-tune models, and apply RLHF to improve agent decisions.

You will deploy AI agents using Vertex AI, PaLM, and Gemini, collaborating with cross-functional teams to define architecture, orchestration, and evaluation frameworks. Proficient in Python and major ML frameworks is essential.

Qualifications

• Minimum 3 years of experience in AI/ML engineering.
• Strong understanding of Large Language Models (LLMs).
• Experience with prompt engineering and prompt optimization.
• Hands-on experience with RAG architectures.
• Experience fine-tuning foundation models using domain-specific datasets.
• Practical experience applying RLHF techniques.
• Knowledge of transformer-based architectures.

Responsibilities

• Design, develop, and optimize prompt engineering strategies for large language models.
• Build and curate high-quality datasets for RAG and model fine-tuning.
• Fine-tune large language models using domain-specific datasets.
• Apply RLHF techniques to improve model accuracy and agent decision-making.
• Design, develop, and deploy AI agents using Vertex AI, PaLM, and Gemini.
• Collaborate with engineering teams to define agent architecture, orchestration strategies, and evaluation frameworks.
• Implement Agent-to-Agent (A2A) communication and orchestration mechanisms.
• Standardize agent interactions using the MCP.
• Develop and maintain scalable machine learning pipelines.
• Evaluate model performance and implement continuous optimization strategies.
• Conduct experiments involving prompt tuning, chain-of-thought reasoning, and agent workflows.
• Stay current with the latest developments in generative AI, agent frameworks, and LLM optimization techniques.

Skills

Prompt Engineering

LLM Fine-Tuning

RAG Dataset Curation

RLHF

Agent Architecture

A2A Protocols

MCP

Vertex AI

PaLM

Gemini

Python

TensorFlow

JAX

PyTorch

LoRA

PEFT

Quantization

Prompt Tuning

Chain-of-Thought Reasoning

Tools

Vertex AI

PaLM

Gemini

ADK

Agentspace

Model Context Protocol (MCP)

A2A Protocols

Job description

Tech Aalto Pte ltd seeks an experienced AI/ML Engineer to design, implement, and optimize prompt strategies for cutting-edge large language models. You will build RAG datasets, fine-tune models, and apply RLHF to improve agent decisions.

You will deploy AI agents using Vertex AI, PaLM, and Gemini, collaborating with cross-functional teams to define architecture, orchestration, and evaluation frameworks. Proficient in Python and major ML frameworks is essential.