Job Description

We are looking for an AI & Data Engineer to build and scale next-generation intelligent systems. This role goes beyond traditional model development, you will design agentic workflows that can reason, remember, and act autonomously.

You will bridge the gap between data engineering, machine learning, and LLM-based systems, creating robust pipelines and integrating them with modern AI frameworks. This role is ideal for someone who enjoys working at the intersection of AI, data infrastructure, and system architecture.

Key Responsibilities

AI & LLM Orchestration

Design and implement agentic systems capable of multi-step reasoning, planning, and tool usage
Build state and memory management systems for maintaining context in LLM workflows
Implement Model Context Protocol (MCP) to enable structured interaction between models and external tools/data
Develop and optimize RAG (Retrieval Augmented Generation) pipelines using vector databases

Data Engineering & Machine Learning

Write complex analytical SQL queries for data transformation and analysis
Design and maintain data warehouse models (fact and dimension tables)
Build scalable ETL/ELT pipelines to ensure data availability and quality
Perform exploratory data analysis (EDA) to support model development
Implement supervised and unsupervised learning models
Develop solutions such as:
classification and regression models
recommendation systems
clustering and segmentation

Natural Language Processing (NLP)

Implement and fine-tune NLP models for tasks such as:
named entity recognition (NER)
intent classification and translation
Work with transformer-based architectures (e.g., BERT) to improve text understanding

KPIs

Successful implementation of agentic workflows and LLM-based systems
Performance and scalability of data pipelines and ML models
Accuracy and effectiveness of AI-driven features (RAG, NLP, recommendations)
Data quality, reliability, and availability across pipelines
Efficiency of data processing and feature engineering workflows
Contribution to product intelligence and data-driven decision-making

Must-Have Skills & Experience

Strong proficiency in Python (FastAPI, Pydantic, Pandas, Scikit-learn, PySpark)
Hands-on experience with LLM/AI frameworks (e.g., LangChain, LangGraph)
Experience building or working with agentic workflows or AI-driven systems
Strong experience with relational databases (PostgreSQL, Snowflake, BigQuery)
Experience with vector databases (Pinecone, Milvus, Weaviate, pgvector, etc.)
Solid understanding of data warehousing concepts (star/snowflake schema, fact/dimension modeling)
Experience building ETL pipelines (Python, Airflow)
Strong SQL and data modeling skills
Experience with machine learning fundamentals (classification, regression, clustering)
Experience working with NLP models and Hugging Face ecosystem

Nice-to-Have Skills

Experience with graph databases (e.g., Neo4j) for GraphRAG or knowledge graphs
Familiarity with MLOps/DevOps practices (Docker, Kubernetes, CI/CD)
Experience with DBT
Understanding of JSON-RPC or MCP architecture
Experience building advanced RAG pipelines or AI applications in production

Traits

Strong systems thinker with ability to connect data, AI, and infrastructure
Curious and experimental mindset, especially in AI and agentic systems
High attention to detail in data quality and model performance
Comfortable working in a fast-paced, evolving technical environment
Strong problem solver with ability to work across multiple domains (AI, data, backend)

What We Offer:

Competitive salary and benefits.
A dynamic and supportive work environment.
Opportunities for professional growth and development.
The chance to work on cutting-edge technologies and projects.

Who we are:

Wolkk is an offshore outsourcing company dedicated to connecting international clients with top talent in Indonesia. Our mission is to enable young professionals in Indonesia to learn and grow by working with international clients. We help clients recruit and manage their employees in Indonesia, fostering an environment where talent can thrive and businesses can achieve their goals. Join us at Wolkk and be part of a dynamic team that bridges global opportunities with local expertise.

Our values: