BJAK

Machine Learning Engineer, LLM

BJAK

full-time

Posted on:

Location Type: Remote

Location: Remote • 🇻🇳 Vietnam

Visit company website
AI Apply
Apply

Job Level

Mid-LevelSenior

Tech Stack

PyTorchTensorflow

About the role

  • Fine-tune & adapt - Use LoRA/QLoRA to optimize open-source models (LLaMA, Mistral, Gemma)
  • Engineer prompts & curates data - Craft prompts and datasets that reflect tone, brand voice, and safety.
  • Evaluate models – Build metrics pipelines for perplexity, toxicity, and relevance to ensure safe and high-quality outputs.
  • Deploy & monitor – Scale models into production with performance optimization and monitoring for drift.
  • Collaborate & deliver – Partner with product, engineering, and design teams to launch user-facing AI features.

Requirements

  • Strong experience in transformers, deep learning, and fine-tuning methods (LoRA/QLoRA, SFT, distillation).
  • Proficiency with PyTorch (preferred) or TensorFlow.
  • Skilled in prompt engineering and dataset curation for alignment with tone, safety, and trust.
  • Familiar with evaluation metrics: perplexity, toxicity, relevance.
  • Strong software engineering foundations in algorithms, data structures, and clean code practices.
Benefits
  • Flat structure & real ownership
  • Full involvement in direction and consensus decision making
  • Flexibility in work arrangement
  • High-impact role with visibility across product, data, and engineering
  • Top-of-market compensation and performance-based bonuses
  • Global exposure to product development
  • Lots of perks - housing rental subsidies, a quality company cafeteria, and overtime meals
  • Health, dental & vision insurance
  • Global travel insurance (for you & your dependents)
  • Unlimited, flexible time off

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard skills
LoRAQLoRAtransformersdeep learningPyTorchTensorFlowprompt engineeringdataset curationevaluation metricssoftware engineering
Soft skills
collaborationcommunication
BJAK

MLOps Engineer

BJAK
Mid · Seniorfull-time🇻🇳 Vietnam
Posted: 1 day agoSource: jobs.ashbyhq.com
KubernetesRay
BJAK

Founding AI Engineer

BJAK
Mid · Seniorfull-time🇻🇳 Vietnam
Posted: 1 day agoSource: jobs.ashbyhq.com
PythonPyTorchScikit-LearnTensorflow
BJAK

Lead AI Engineer

BJAK
Seniorfull-time🇻🇳 Vietnam
Posted: 1 day agoSource: jobs.ashbyhq.com
PythonPyTorchScikit-LearnTensorflow
BJAK

Machine Learning Engineer – Diffusion, Vision

BJAK
Mid · Seniorfull-time🇻🇳 Vietnam
Posted: 1 day agoSource: jobs.ashbyhq.com
FluxPyTorch