Skip to main content
BJAK

Staff Machine Learning Engineer

RemoteUnited States only
Published
Role
Unknown
Experience
Staff
Employment
Full-time
Salary not disclosed
Check eligibility

Open to US only. Set where you work from to check your eligibility.

No BS summary

We are building AI-native proactive applications for everyday tasks like email, notes, and calendars, aiming to reduce hallucinations and organize users' lives. As Staff ML Engineer, you will own the execution layer of our intelligence, turning research and model capabilities into reliable, scalable production systems, working across the model lifecycle from data to deployment.

Core skills

Python/PyTorch/JAX

What you'll do

  • Own the end-to-end ML systems powering our company, from data and training to evaluation, inference, and deployment.
  • Build and evolve training and fine-tuning pipelines for large models.
  • Design evaluation systems that measure capability, robustness, safety, and real-world product performance.
  • Architect high-performance inference systems, optimizing latency, GPU utilization, memory, cost, and reliability.
  • Build data pipelines and systems for high-quality real-world and synthetic training data.
  • Establish reliable production infrastructure for deploying, monitoring, and continuously improving models.
  • Partner closely with research and application engineering to turn model capabilities into product improvements.
  • Make pragmatic technical trade-offs and rapidly iterate based on real-world performance.

What they require

  • Experience building and shipping ML systems used in production, not just research prototypes.
  • Strong understanding of modern large-model training, fine-tuning, evaluation, and inference.
  • Strong software engineering and systems fundamentals.
  • Experience operating ML workloads at meaningful scale, particularly GPU-based systems.
  • Strong technical judgment and the ability to navigate ambiguous problems independently.
  • A bias toward experimentation, measurement, and shipping.
  • High standards for correctness, reliability, and production quality.

AI Neobank (financial services) building mobile banking experiences

🇭🇰 Hong Kong SAR ChinaFintechStartupbjak.my/

What people say about this company

2.6/ 5

Salary not disclosed