BJAK Logo

BJAK

Technical Lead, Machine Learning

Reposted One Month Ago
Be an Early Applicant
In-Office or Remote
Hiring Remotely in Singapore, SGP
Senior level
In-Office or Remote
Hiring Remotely in Singapore, SGP
Senior level
Lead execution of end-to-end ML systems: data pipelines, training workflows, evaluation, and inference. Translate research into production by fine-tuning large models (LoRA/QLoRA/SFT/DPO/distillation), designing scalable GPU inference, ensuring memory/latency optimization, building evaluation for performance/robustness/safety, and integrating ML into backend and client products while operating under real-world constraints.
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 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

5 Hours Ago
Remote or Hybrid
HarbourFront, SGP
Entry level
Entry level
eCommerce • Fashion • Retail • Sales • Wearables • Design
Provide personalized styling advice, product recommendations, and exceptional customer service in a Kate Spade retail environment. Engage customers, drive sales through storytelling and add-on suggestions, complete POS transactions, maintain stockroom organization, and support omni-channel selling. The role requires flexible scheduling, teamwork, strong communication, retail experience, and knowledge of brand products.
Top Skills: Omni-Channel SellingPos SystemsSocial Media
Yesterday
Easy Apply
Remote or Hybrid
Singapore, SGP
Easy Apply
Senior level
Senior level
Cloud • Information Technology • Security • Software • Cybersecurity
Provision, stage, troubleshoot, repair, and maintain bare-metal servers, network devices, and data center infrastructure. Collaborate with engineering and installation teams to validate builds, improve deployment workflows, and optimize staging processes for a global cloud security network. The role requires expertise in hardware configurations, physical connectivity, Linux/Unix, BIOS, out-of-band management, virtualization, scripting, and build automation. Occasional travel to a Singapore integration center is required.
Top Skills: Ai/MlAnsibleBashBiosConfluenceIdracIloIpmiJIRAKvmLinux/UnixProxmoxPxePythonSaa
Yesterday
Remote or Hybrid
Singapore, SGP
Senior level
Senior level
Cloud • Computer Vision • Information Technology • Sales • Security • Cybersecurity
Deploys, configures, optimizes, and supports CrowdStrike Falcon’s AI-powered detection and response capabilities. The role develops automation playbooks and API integrations, tunes machine-learning detection models, troubleshoots customer issues, supports threat hunting and forensic analysis, integrates security workflows, and leads AI workshops. It also requires customer engagement, technical presentations, project prioritization, cross-functional leadership, and mentoring while applying cybersecurity, AI/ML, and platform deployment expertise.
Top Skills: Active DirectoryArtificial IntelligenceAWSAzureAzure AdBashChefCitrixCrowdstrike FalconGCPGitGroup PolicyIbm BigfixLinuxLogscaleMachine LearningmacOSMcafee EpoMssqlPdqPowershellPsexecPuppetPythonRRest ApisSccmSIEMSplunkSsoWindows

What you need to know about the Singapore Tech Scene

The digital revolution has driven a constant demand for tech professionals across industries like software development, data analytics and cybersecurity. In Singapore, one of the largest cities in Southeast Asia, the demand for tech talent is so high that the government continues to invest millions into programs designed to develop a talent pipeline directly from universities while also scaling efforts in pre-employment training and mid-career upskilling to expand and elevate its workforce.

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account