BJAK Logo

BJAK

Applied Machine Learning Engineer

Reposted One Month Ago
Remote
Hiring Remotely in United States
Senior level
Remote
Hiring Remotely in United States
Senior level
The Senior Machine Learning Engineer will build and own production ML systems, manage end-to-end workflow, debug issues, and mentor others.
The summary above was generated by AI
About

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.

 
Role

Build and improve ML systems that power our AI-native applications.

Work on classification, retrieval, ranking, personalization and evaluation, taking ownership from model development to production deployment.

This role is for engineers who enjoy solving practical ML problems, improving model quality and shipping reliable systems used by real users.

 
Focus
  • Build and improve ML models for classification, ranking, retrieval and personalization.

  • Develop efficient classifiers and optimize model accuracy, precision, latency and cost.

  • Improve search relevance and context retrieval using embeddings, hybrid search and reranking.

  • Build personalization models that learn from user behavior while respecting privacy.

  • Develop evaluation pipelines, labeled datasets and metrics to measure model quality.

  • Fine-tune and adapt existing models when supported by sufficient data.

  • Build and maintain ML training, inference and deployment pipelines.

  • Debug production issues, monitor model performance and ship improvements based on real-world feedback.

 
Tech Stack
  • Python

  • PyTorch / JAX

  • Embeddings, vector search and ranking models

  • Production ML systems and inference pipelines

 
Ideal Experience
  • Strong foundations in machine learning and modern neural architectures.

  • Hands-on experience building and deploying ML models in production.

  • Experience in classification, ranking, retrieval, recommendation systems or personalization.

  • Understanding of model evaluation, experimentation and performance metrics.

  • Comfortable writing production-quality code and working with real-world datasets.

  • Able to balance model quality, latency, cost and reliability.

  • Strong ownership, problem-solving skills

 
Outcomes
  • ML models in production meet expected accuracy, relevance, latency and reliability targets.

  • Classification and ranking improvements deliver measurable gains in model quality.

  • Evaluation pipelines identify regressions and support reliable model releases.

  • Training, inference and data pipelines are robust, reproducible and maintainable.

  • Model improvements are driven by real-world feedback and measurable results.

  • Collaborates effectively with engineering and product teams to deliver reliable ML-powered features.

 
How We Work

The best products in the world are built by small, world-class teams. We are a high-talent-density, hands-on team. We make decisions collectively and move quickly, balancing high-quality work with continuous learning.

Joining our team requires the ability to bring structure, exercise judgment and execute independently. Our goal is to put a truly magical product into the hands of our users.

 
Interview Process

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

Applications are evaluated by our technical team members. Interviews will be conducted virtually 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.

Similar Jobs

9 Days Ago
In-Office or Remote
United States
185K-245K Annually
Junior
185K-245K Annually
Junior
Information Technology • Professional Services • Software • Cybersecurity
Own machine learning models end to end, from data preparation and training through evaluation and production deployment. Build pipelines for large multimodal datasets including video, sensor streams, and language. Collaborate with engineering and computer vision teams, rapidly prototype solutions, apply relevant research, and continuously improve production models and tooling.
Top Skills: Computer VisionDeep LearningJaxPythonPyTorch
24 Days Ago
In-Office or Remote
Entry level
Entry level
Information Technology
Build end-to-end machine learning systems spanning research reproduction, model evaluation, inference infrastructure, backend services, and product interfaces. Design rigorous experiments, evaluation datasets, and verification workflows; analyze model internals; investigate provenance and evasion; and convert findings into reliable production systems with APIs, background jobs, observability, testing, documentation, and user-facing React/TypeScript experiences.
Top Skills: DspyGpusHugging Face TransformersLitellmNext.JsPgvectorPostgresPythonPyTorchRayReactTemporalTypescriptVllm
One Month Ago
Remote
IL, USA
120K-194K Annually
Senior level
120K-194K Annually
Senior level
Insurance
Builds and maintains full-stack software products and backend services with a strong Java and Spring Boot focus. Develops APIs, UI, infrastructure, CI/CD pipelines, and production machine learning or GenAI systems. Practices test-driven development, supports incident response, addresses security and compliance findings, performs root-cause analysis, and coaches junior engineers. Collaborates on architecture, participates in Agile ceremonies, and improves observability, deployment readiness, and product quality.
Top Skills: AWSAzureGCPGithub ActionsJava 17+JenkinsMongoDBOauthOidcPythonSpring Boot 4.X+Sso

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