BJAK Logo

BJAK

Technical Lead, Machine Learning

Posted One Month Ago
Be an Early Applicant
Remote or Hybrid
Hiring Remotely in United States
Senior level
Remote or Hybrid
Hiring Remotely in United States
Senior level
Lead the end-to-end ML execution: build and operate data pipelines, training and evaluation systems, and scalable inference. Fine-tune and optimize large models (LoRA/QLoRA/SFT/DPO/distillation), ensure GPU efficiency, monitor production quality, collaborate with product and application engineering, and rapidly ship reliable, safe ML solutions that translate research into production.
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 product focuses on achieving high reliability for long-running workflows, persistent context, and real-world task completion. We believe products will greatly reduce hallucinations

Our objective is to organise anyone's life, allowing us all to spend time on valuable and meaningful things

As Technical Lead, 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

27 Days Ago
Remote
United States
204K-276K Annually
Expert/Leader
204K-276K Annually
Expert/Leader
Aerospace • Information Technology • Professional Services • Security • Software
Lead AI/ML strategy, architecture, development, deployment, and operationalization for CMS mission environments. Design predictive analytics, automation, and decision-support systems; oversee data pipelines, MLOps, model validation, explainability, and bias mitigation. Integrate secure AI solutions with cybersecurity, governance, compliance, and risk management requirements. Develop RAG and agentic AI capabilities, transition prototypes to production, advise stakeholders, support audits and governance reviews, and mentor engineering teams.
Top Skills: AgentforceAgentic AiAWSAws BedrockAws GovcloudAws SagemakerAzureAzure Ai FoundryAzure GovernmentGCPGoogle Vertex AiJavaMlopsPythonPyTorchRRetrieval-Augmented Generation (Rag)SalesforceTensorFlowZero Trust
23 Days Ago
In-Office or Remote
255K-345K Annually
Senior level
255K-345K Annually
Senior level
eCommerce • Mobile • Retail
Lead the Discovery Platform team, overseeing the development of scalable, high-performance systems for retrieval and ranking to enhance the user experience in Whatnot's live social marketplace.
Top Skills: Aws SagemakerEc2EcsEksElasticsearchFlinkKafkaKinesisLambdaLuceneOpensearchS3SolrSpark
12 Minutes Ago
Easy Apply
Remote or Hybrid
11 Locations
Easy Apply
80K-95K Annually
Senior level
80K-95K Annually
Senior level
Edtech • Kids + Family • Social Impact • Software
Leads learning design for K-8 digital education content initiatives from brief through launch. Develops instructional frameworks, assessment items, content specifications, quality standards, and grade-band differentiation. Provides feedback and upskills learning designers, partners with product and editorial teams, and contributes to AI-assisted content workflows through prompt design, output evaluation, and quality rubrics. Requires strong assessment writing, curriculum expertise, standards knowledge, initiative leadership, and experience designing for multilingual learners.
Top Skills: Ai-Assisted Content WorkflowsArtificial IntelligenceCcssCmsEld StandardsNgssPrompt DesignWida

What you need to know about the Austin Tech Scene

Austin has a diverse and thriving tech ecosystem thanks to home-grown companies like Dell and major campuses for IBM, AMD and Apple. The state’s flagship university, the University of Texas at Austin, is known for its engineering school, and the city is known for its annual South by Southwest tech and media conference. Austin’s tech scene spans many verticals, but it’s particularly known for hardware, including semiconductors, as well as AI, biotechnology and cloud computing. And its food and music scene, low taxes and favorable climate has made the city a destination for tech workers from across the country.

Key Facts About Austin Tech

  • Number of Tech Workers: 180,500; 13.7% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Dell, IBM, AMD, Apple, Alphabet
  • Key Industries: Artificial intelligence, hardware, cloud computing, software, healthtech
  • Funding Landscape: $4.5 billion in VC funding in 2024 (Pitchbook)
  • Notable Investors: Live Oak Ventures, Austin Ventures, Hinge Capital, Gigafund, KdT Ventures, Next Coast Ventures, Silverton Partners
  • Research Centers and Universities: University of Texas, Southwestern University, Texas State University, Center for Complex Quantum Systems, Oden Institute for Computational Engineering and Sciences, Texas Advanced Computing Center

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account