Machine Learning Engineer

PT Prima Vista Solusi; Jakarta Pusat; On-site; IDR 350,000,000 - 750,000,000; Full time

Job summary

PT Prima Vista Solusi in Jakarta is seeking a seasoned ML/NLP engineer to design, develop, and deploy cutting-edge LLMs for diverse NLP applications in Jakarta. You will collaborate with data scientists, software engineers, and product managers to integrate AI in production systems and optimize large-scale models.

Responsibilities include fine-tuning transformers, applying reinforcement learning techniques, building scalable ML pipelines, and ensuring responsible AI practices.

Qualifications

• Bachelor’s or Master’s in CS/AI/ML or related field.
• 3–5 years in ML/DL/NLP roles.
• Proficient in Python and ML frameworks.
• Experience with large-scale LLMs and transformers.

Responsibilities

• Design, develop, and deploy cutting-edge LLMs for NLP applications.
• Fine-tune models with TensorFlow, PyTorch, or JAX.
• Research advancements in LLMs and transformer architectures.
• Collaborate with data scientists, engineers, and PMs to productionize AI.
• Build scalable ML pipelines for training and inference.
• Ensure efficiency, performance, and scalability of models.
• Evaluate AI models with NLP metrics and refine based on results.
• Maintain up-to-date knowledge and contribute to open-source.
• Apply responsible AI practices including fairness and explainability.
• Document architectures, training methods, and deployment strategies.

Skills

Python

Large-scale LLMs

Transformers

Reinforcement Learning

Cloud Platforms

MLOps

Distributed Computing

Data Preprocessing

Collaboration

Prompt Engineering

Education

Bachelor's or Master's in CS/AI/ML

Tools

TensorFlow

PyTorch

JAX

AWS

Google Cloud

Azure

Spark

Ray

Job description

• Design, develop, and deploy cutting-edge LLMs for diverse NLP applications.
• Optimize and fine-tune large-scale language models using frameworks like TensorFlow, PyTorch, or JAX.
• Research and implement advancements in LLMs, including transformer architectures and reinforcement learning techniques.
• Work closely with data scientists, software engineers, and product managers to integrate LLMs into production systems.
• Build and maintain scalable machine learning pipelines for training and inference.
• Ensure LLMs are efficient, high-performing, and scalable for real-world applications.
• Evaluate AI models using key NLP metrics and refine them based on experimental results.
• Stay updated with the latest AI research and contribute to open-source projects where relevant.
• Implement responsible AI principles, including fairness, explainability, and ethical considerations.
• Document AI model architectures, training methodologies, and deployment strategies.

Minimum Qualifications

• Bachelor's, Master's in Computer Science, Artificial Intelligence, Machine Learning, or a related field.
• Minimum 3-5 years of experience in machine learning, deep learning, or natural language processing roles.
• Strong proficiency in Python and experience with ML frameworks like TensorFlow, PyTorch, or JAX.
• Hands-on experience with large-scale language models, including transformers, GPT-based models, and BERT-style architectures.
• Experience with cloud platforms (AWS, Google Cloud, Azure) and ML model deployment.
• Knowledge of MLOps practices, including model versioning, monitoring, and automation.
• Experience with large-scale datasets, data preprocessing, and distributed computing frameworks (e.g., Spark, Ray).
• Solid understanding of deep learning architectures, optimization techniques, and reinforcement learning.
• Strong problem-solving skills and ability to conduct independent research.
• Excellent collaboration and communication skills for cross-functional teamwork.
• Experience with prompt engineering, fine-tuning LLMs, and retrieval-augmented generation (RAG) is a must.
• Contributions to AI research publications or open-source projects are highly desirable.