BJAK Logo

BJAK

Principal Machine Learning Engineer

Reposted One Month Ago
Remote or Hybrid
Hiring Remotely in United States
Expert/Leader
Remote or Hybrid
Hiring Remotely in United States
Expert/Leader
Design, build, and operate large-scale ML systems across data, training, evaluation, inference, and deployment. Lead GPU-optimized training and inference pipelines, ensure reliability, performance, safety, and seamless integration into products. Own production deployments, monitoring, and cross-functional best practices to drive measurable model improvements and efficient research-to-production cycles.
The summary above was generated by AI
About the Role

There are over 5 billion users using basic applications today such email, notes, tasks, calendar and they're not AI-native. Our mission is to build proactive applications for anyone in the world, who are not used to complex prompting. We aim to bring intelligence to conversations, errands, organising and workflows, with minimal to no prompting.

Our first application is an agent-first mobile communication app. We believe building applications that users use daily for communication, coordination and collaboration needs will help integrate AI into their lives. The net effect is to eliminate misinformation, achieve low coordination costs, enable high quality tasks and collaboration at scale with minimal or little prompting.

As Principal Machine Learning Engineer, you own the execution layer of our intelligence, turning research and model capabilities into reliable, scalable production systems.

You will work across the model lifecycle: data, training, evaluation, inference, and deployment. This is a hands-on leadership role for someone who wants to operate at the intersection of research, systems, and product.

 
What You'll Own
  • 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 We're Looking For
  • 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.

 
Outcomes
  • Research and models reliably translate into production-ready solutions with clear performance and quality targets.

  • ML pipelines, training loops, and inference systems are stable, efficient, and maintainable.

  • Production issues are detected, debugged, and resolved quickly, minimizing user impact.

  • Team members are supported, aligned, and able to deliver high-impact ML work with minimal friction.

  • Iterations on models and systems are measurable, safe, and improve user experience over time.

 
Tech Stack
  • Python

  • PyTorch / JAX

  • GPU-based training and inference system

 
Ideal Experience
  • You have built or shipped real ML systems used by people, not just demos.

  • You are comfortable working with large models and understanding their failure modes.

  • You write strong, production-grade code and care about system correctness.

 
How We Work

We are a small, high-talent-density, hands-on team. Engineers have broad ownership and are expected to exercise strong judgment and execute independently.

We make decisions quickly, work closely together, and balance speed with engineering fundamentals. We care less about process and more about building something exceptional.

 
Interview process

If there appears to be a fit, we'll reach to schedule 3, but no more than 4 interviews.

Applications are evaluated by our technical team members. Interviews will be conducted via virtual meetings and/or onsite.

We value transparency and efficiency, so expect a prompt decision. If you've demonstrated the exceptional skills and mindset we're looking for, we'll extend an offer to join us. This isn't just a job offer; it's an invitation to be part of a team that's bringing AI to have practical benefits to billions globally.

Similar Jobs

17 Days Ago
In-Office or Remote
185K-317K Annually
Senior level
185K-317K Annually
Senior level
Artificial Intelligence • Big Data • Healthtech • Information Technology • Machine Learning • Software • Analytics
Leads architecture and hands-on delivery of production-grade software, machine learning, generative AI, RAG, and AI-agent solutions. Builds cloud-native services, APIs, data integrations, MLOps/LLMOps pipelines, evaluation frameworks, and secure deployment patterns. Owns technical initiatives from experimentation through production, including reliability, observability, testing, incident resolution, responsible AI, and cost optimization. Influences cross-team architecture and roadmaps while mentoring engineers without direct management responsibilities.
Top Skills: Ai AgentsAmazon Web ServicesAPIsAzureCi/CdContainersGenerative AiGCPInfrastructure As CodeKubernetesLarge Language ModelsLlmopsMlopsPythonPyTorchRetrieval-Augmented GenerationScikit-LearnTensorFlowVector Databases
11 Days Ago
In-Office or Remote
IN, USA
196K-309K Annually
Senior level
196K-309K Annually
Senior level
Cloud • Information Technology • Productivity • Security • Software • App development • Automation
Lead the design and delivery of knowledge graph and machine learning systems that infer personal work context from connected tools. Build graph inference pipelines, schemas, permission-aware APIs, Rovo Chat integrations, and MCP-compatible CLI experiences. Establish evaluation frameworks, improve retrieval and grounding, provide technical direction across teams, mentor engineers, and ensure responsible, privacy-safe AI systems operate reliably at production scale.
Top Skills: Amazon NeptuneCypherDglGraph DatabasesGraph Neural NetworksGraphxKnowledge GraphsLlm OrchestrationMcpNeo4JOntology DesignPygRagRetrieval-Augmented GenerationSemantic EmbeddingsSparql
Yesterday
In-Office or Remote
US
285K-450K Annually
Entry level
285K-450K Annually
Entry level
Social Media
Sets technical vision for Pinterest’s ML and media infrastructure, leading distributed training, fine-tuning, evaluation, inference, GPU optimization, content ingestion, and signal-processing architectures. Partners with AI/ML, Ads, Content, Shopping, and GenAI teams to productionize reliable, scalable platforms, drive architecture decisions and migrations, improve operational standards, and mentor senior engineers.
Top Skills: Ads RankingC++Content IngestionDistributed Ml SystemsFeature PlatformsGpu ComputingInference InfrastructureJavaLlmsModel CompilationModel ServingModel TrainingPythonQuantizationReal-Time Data ProcessingRecommender SystemsSearch SystemsVlms

What you need to know about the Los Angeles Tech Scene

Los Angeles is a global leader in entertainment, so it’s no surprise that many of the biggest players in streaming, digital media and game development call the city home. But the city boasts plenty of non-entertainment innovation as well, with tech companies spanning verticals like AI, fintech, e-commerce and biotech. With major universities like Caltech, UCLA, USC and the nearby UC Irvine, the city has a steady supply of top-flight tech and engineering talent — not counting the graduates flocking to Los Angeles from across the world to enjoy its beaches, culture and year-round temperate climate.

Key Facts About Los Angeles Tech

  • Number of Tech Workers: 375,800; 5.5% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Snap, Netflix, SpaceX, Disney, Google
  • Key Industries: Artificial intelligence, adtech, media, software, game development
  • Funding Landscape: $11.6 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Strong Ventures, Fifth Wall, Upfront Ventures, Mucker Capital, Kittyhawk Ventures
  • Research Centers and Universities: California Institute of Technology, UCLA, University of Southern California, UC Irvine, Pepperdine, California Institute for Immunology and Immunotherapy, Center for Quantum Science and Engineering

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account