BJAK Logo

BJAK

Applied Machine Learning Engineer

Reposted 2 Months 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

2 Days Ago
Remote or Hybrid
Internship
Internship
Artificial Intelligence • Information Technology • Software
AI/ML engineering intern will help design, build, evaluate, and deploy LLM-powered applications and machine learning systems. Responsibilities include developing data pipelines, working with structured and unstructured data, supporting model training and optimization, preparing systems for production monitoring, and collaborating with engineers, researchers, and product teams. The role offers mentorship and hands-on experience applying AI to healthcare, government, and energy workflows.
Top Skills: Generative AiJaxLarge Language Models (Llms)PythonPyTorchRetrieval-Augmented Generation (Rag)TensorFlow
3 Days Ago
Remote
United States
Senior level
Senior level
Artificial Intelligence • Information Technology • Software
Design and build production-grade AI memory systems, including retrieval pipelines, storage tiers, privacy filtering, memory extraction, adaptive updates, hybrid search, and evaluation frameworks. Ship and maintain LLM-powered features, diagnose RAG and prompt failures, implement agentic workflows and tool integrations, and ensure reliable degradation and fallback strategies. Collaborate with product management to define feasible memory-system capabilities while owning production reliability.
Top Skills: Anthropic ApiChromaCohere ApiDockerElasticsearchFastapiGitGitGithub ActionsHaystackJavaScriptJSONLangchainLanggraphLettaLlamaindexMcpMem0MemgptMongoDBOpenai ApiPineconePostgresPythonQdrantTypescriptWeaviate
6 Days Ago
In-Office or Remote
Austin, TX, USA
80K-400K Annually
Senior level
80K-400K Annually
Senior level
Digital Media • Information Technology • Professional Services • Consulting
Leads client-facing discovery and technical product development for AI initiatives. Designs, prototypes, deploys, and operates agentic systems, retrieval pipelines, enterprise integrations, and production-grade generative AI applications. Establishes evaluation, observability, safety, governance, and reliability frameworks while translating business needs into technical specifications. Requires strong Python and software engineering expertise, cloud and DevOps experience, retrieval architecture knowledge, and at least seven years of engineering experience, including production generative AI experience.
Top Skills: AWSAzureCi/CdClaude CodeCodexCursorDockerGCPGithub CopilotKubernetesLoraMcpOpencodePeftPythonRagRestWebsockets

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