Founding Machine Learning Engineer
BJAK
full-time
Posted on:
Location Type: Remote
Location: Remote • 🇺🇸 United States
Visit company websiteJob 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