BJAK

Founding AI/ML Research Engineer

BJAK

full-time

Posted on:

Location Type: Remote

Location: Remote • 🇰🇷 South Korea

Visit company website
AI Apply
Apply

Job Level

Mid-LevelSenior

Tech Stack

PyTorchTensorflow

About the role

  • Fine-tune and adapt models as needed while also designing new model architectures and training pipelines from scratch
  • Design and curate datasets, craft prompts, and experiment with alignment and safety strategies for high-quality outputs.
  • Build evaluation pipelines for model performance, safety, and alignment, and innovate on new metrics for real-world impact.
  • Ship models to production, optimise for latency and scale, and monitor drift or degradation
  • Collaborate with product, engineering, and design to launch polished, user-facing AI features
  • Explore new modeling approaches, optimisation methods, and product use cases

Requirements

  • Strong experience with transformers, deep learning, and fine-tuning methods (LoRA/QLoRA, SFT, distillation)
  • Proficiency in PyTorch (preferred) or TensorFlow
  • Experience in prompt engineering and dataset creation for alignment, tone, trust, and safety
  • Familiarity with evaluation methods: perplexity, toxicity, relevance, robustness
  • Solid software engineering fundamentals — algorithms, data structures, clean code
  • Ability to operate in a fast, lean, and high-ownership environment
Benefits
  • High ownership and autonomy from day one
  • Direct involvement in technical direction and product decisions
  • Remote-first flexibility
  • A high-impact role with visibility across ML, engineering, and product
  • Competitive compensation and performance-based bonuses
  • Backing of a profitable US$2B group, with the speed of a startup
  • Insurance coverage, flexible time off, and global travel insurance
  • Opportunity to shape a new global AI product from zero

Applicant Tracking System Keywords

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

Hard skills
transformersdeep learningfine-tuning methodsLoRAQLoRASFTdistillationPyTorchTensorFlowprompt engineering
Soft skills
collaborationfast-paced environmentownership