BJAK Logo

BJAK

Lead Engineer, Machine Learning

Reposted One Month Ago
Remote or Hybrid
Hiring Remotely in United States
Expert/Leader
Remote or Hybrid
Hiring Remotely in United States
Expert/Leader
Leads the end-to-end execution layer for machine learning, including data pipelines, large-model training and fine-tuning, evaluation, inference, deployment, monitoring, and reliability. Designs scalable GPU-based systems, optimizes performance and cost, partners with research and application engineering, and translates model capabilities into measurable production improvements. Also provides technical leadership, judgment, and support for high-impact ML delivery.
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 Lead Engineer, Machine Learning, 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

3 Days Ago
In-Office or Remote
146K-250K Annually
Senior level
146K-250K Annually
Senior level
Artificial Intelligence • Big Data • Healthtech • Information Technology • Machine Learning • Software • Analytics
Leads the design, development, and deployment of AI and machine learning solutions for healthcare technology platforms. Builds scalable ML models, MLOps pipelines, LLM and RAG systems, backend services, and agent-based applications. Establishes standards for monitoring, testing, CI/CD, and lifecycle management while collaborating with engineering, data, product, security, and SRE teams. Provides architectural guidance and technical mentorship, evaluates emerging AI technologies, and promotes responsible, secure, and resilient AI implementation.
Top Skills: AWSAzureAzure Ai SearchCi/CdCrewaiDockerEmbeddingsFaissGoogle Cloud PlatformJavaJavaScriptLangchainLanggraphLarge Language ModelsLlamaindexMcpMilvusMlopsModel Fine-TuningPeftPgvectorPineconePrompt EngineeringPythonRetrieval-Augmented GenerationSemantic Kernel
3 Days Ago
In-Office or Remote
146K-250K Annually
Senior level
146K-250K Annually
Senior level
Artificial Intelligence • Big Data • Healthtech • Information Technology • Machine Learning • Software • Analytics
Leads the design, development, deployment, and operationalization of enterprise AI solutions, including LLM applications, RAG systems, intelligent agents, classification models, and decision-support tools. Establishes prompt strategies, guardrails, evaluation frameworks, monitoring, observability, and governance practices. Partners with Finance, Data Engineering, Architecture, Security, and Responsible AI teams to deliver scalable, secure, compliant, and production-ready solutions.
Top Skills: Ai AgentsAi ObservabilityAPIsArtificial IntelligenceCloud-Native ArchitecturesData PlatformsLarge Language Models (Llms)Machine LearningModel EvaluationModel MonitoringPrompt EngineeringRetrieval-Augmented Generation (Rag)Semantic SearchVector Databases
26 Days Ago
Remote or Hybrid
OH, USA
Senior level
Senior level
Financial Services
Leads enterprise adoption of AI, machine learning, generative AI, and intelligent automation across federated engineering teams. Defines reference architectures, standards, reusable patterns, and Responsible AI governance while advising delivery teams on secure, resilient, scalable implementations. Influences senior stakeholders, promotes AI-assisted engineering practices, establishes validation standards, and builds communities of practice to improve consistency, delivery speed, and operational readiness.
Top Skills: Ai And Machine LearningAi-Assisted Software Development ToolsAPIsCloud ArchitectureGenerative AiIntelligent AutomationLarge Language Models (Llms)MicroservicesNatural Language Processing (Nlp)Responsible AiSmall Language Models (Slms)Software Development Life Cycle (Sdlc)

What you need to know about the San Francisco Tech Scene

San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.

Key Facts About San Francisco Tech

  • Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Google, Apple, Salesforce, Meta
  • Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
  • Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
  • Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account