BJAK is looking for a Machine Learning Engineer (LLM) to join our team. In this role, you will fine-tune state-of-the-art models, design evaluation frameworks, and bring AI features into production, ensuring models are safe, trustworthy, and impactful at scale. You will work closely with regional teams across product, engineering, operations, infrastructure, and data.
What You'll Do
- Fine-tune and adapt open-source models like LLaMA, Mistral, and Gemma using LoRA/QLoRA.
- Engineer prompts and curate datasets that reflect tone, brand voice, and safety.
- Build metrics pipelines for perplexity, toxicity, and relevance to evaluate models.
- Scale models into production with performance optimization and monitoring for drift.
- Partner with product, engineering, and design teams to launch user-facing AI features.
What We're Looking For
- Strong experience in transformers, deep learning, and fine-tuning methods like LoRA/QLoRA, SFT, and 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.
Nice to Have
- Prior work in text generation, moderation, or personalization.
- Experience with RLHF or reinforcement learning in LLMs.
- Contributions to open-source ML projects.
Technical Stack
- PyTorch
- TensorFlow
- LoRA/QLoRA
Team & Environment
We operate with a flat structure with full involvement in direction and consensus decision making. You will work closely with regional teams across product, engineering, operations, infrastructure and data.
Benefits & Compensation
- Flat structure and real ownership.
- 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.
- Housing rental subsidies.
- Quality company cafeteria, and overtime meals.
- Health, dental and vision insurance.
- Global travel insurance for you and your dependents.
- Unlimited, flexible time off.
Work Mode
This is a fully remote position open to candidates globally.
BJAK is an equal opportunity employer.
