BJAK
Company
VP of Research, Machine Learning
Job Description
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.
BJAK
235 jobs posted
About the job
Similar Jobs
Discover more opportunities that match your interests

BJAK
23 hours agoVP of Research, Machine Learning
ChinaView detailsHightouch
3 days agoHead of Machine Learning
Remote$230K - $350K/yrView detailsTikTok
30 days agoMachine Learning Algorithm Research Engineer
SingaporeView detailsTikTok
26 days agoMachine Learning Algorithm Research Engineer
SingaporeView detailsTikTok
8 days agoMachine Learning Algorithm Research Engineer
SingaporeView details
EarnIn
3 days agoHead of Machine Learning, India
Bengaluru, IndiaView detailsTikTok
2 days agoMachine Learning Algorithm Research Engineer
SingaporeView detailsAnthropic
9 days agoMachine Learning Systems Engineer, Research Tools
San Francisco, CA | New York City, NY | Seattle, WA$320K - $405K/yrView detailsReddit
22 days agoDirector of Machine Learning, Safety & Mods
Remote$266K - $365K/yrView details
BJAK
23 hours agoMember of Technical Staff, Machine Learning
ChinaView details
Looking for something different?
Browse all AI jobs