BJAK Logo

BJAK

VP of Research, Machine Learning

Reposted 4 Days Ago
Be an Early Applicant
Remote
Hiring Remotely in United States
Expert/Leader
Remote
Hiring Remotely in United States
Expert/Leader
The VP of Research, Machine Learning will set research direction for AI systems, focusing on context representation, reasoning, and evaluation frameworks while ensuring product safety and alignment.
The summary above was generated by AI
About the Role

A1 is building a proactive AI system that goes beyond responding to prompts – it maintains context across conversations, plans actions, and carries work forward over time.

You will own the research and intelligence direction of this system. Your role is to define how AI reasons, evaluates, and improves in a product used with high frequency.


What You'll Do
  • Set and evolve the research direction for A1’s core intelligence, including context representation, memory, reasoning, planning, and orchestration.

  • Decide when to design new model architectures versus adapting or leveraging frontier open-source or commercial models.

  • Define evaluation frameworks that measure real-world usefulness, robustness, safety, and long-term behavior – not benchmark vanity.

  • Own alignment, safety, and guardrail strategy as first-class product concerns.

  • Guide exploration of frontier techniques such as:

    • retrieval-augmented training

    • mixture-of-experts

    • distillation

    • multi-agent orchestration

    • multimodal systems

  • Shape early product intelligence direction in close partnership with product and application engineering.

  • Set the technical bar for research rigor, judgment, and taste across the organization.


Requirements
  • Deep experience building or evolving real machine learning systems used in production.

  • Strong technical judgment around model behavior, failure modes, and long-horizon trade-offs.

  • A builder’s mindset: you care about systems that work in the real world, not just ideas.

  • Comfortable making irreversible or high-impact decisions with incomplete information.

  • Obsession with evaluation, correctness, and how systems behave over time.

  • High ownership mentality — you operate as a founder, not a manager.

If you are looking to focus primarily on publishing, incremental benchmarks, or managing a large research organization, this role will not be a fit.


Tech Stack
  • Python

  • PyTorch / JAX

  • GPU-based training and inference system


How We Work

Our organization is very flat and our team is small, highly motivated, and focused on engineering and product excellence. All members are expected to be hands-on and to contribute directly to the company’s mission.


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.

Top Skills

Jax
Python
PyTorch

Similar Jobs

2 Hours Ago
Remote
United States
160K-175K Annually
Senior level
160K-175K Annually
Senior level
Software
The Lead Product Designer will shape product direction, contribute to strategy, design high-fidelity UI, and evolve the design system, while collaborating with product and engineering teams.
Top Skills: CSSFigmaHTML
2 Hours Ago
Easy Apply
Remote or Hybrid
USA
Easy Apply
88K-125K Annually
Mid level
88K-125K Annually
Mid level
Cloud • Information Technology • Security • Software • Cybersecurity
Provide technical support for Zscaler products focusing on troubleshooting security and networking issues while ensuring client satisfaction and operational excellence.
Top Skills: DnsHttp/SOidcSAMLTcp/IpTls/Ssl
2 Hours Ago
Remote
USA
194K-235K Annually
Senior level
194K-235K Annually
Senior level
Computer Vision • Healthtech • Information Technology • Logistics • Machine Learning • Software • Manufacturing
The Senior Product Design Manager will lead Clinical Experience and Growth design teams, ensuring high-quality design, team development, and collaboration with cross-functional partners to enhance dentist and patient interactions in dental practices.
Top Skills: Clinical WorkflowsDesign ToolsProduct Design

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