BJAK Logo

BJAK

VP of Research, Machine Learning

Posted Yesterday
Be an Early Applicant
Remote
Hiring Remotely in United States
Expert/Leader
Remote
Hiring Remotely in United States
Expert/Leader
This role leads the research direction for AI, focusing on evolving machine learning systems, aligning product concerns, and ensuring rigorous evaluation of systems.
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

Gpu-Based Training And Inference System
Jax
Python
PyTorch

Similar Jobs

50 Minutes Ago
Remote or Hybrid
USA
145K-220K Annually
Senior level
145K-220K Annually
Senior level
Cloud • Computer Vision • Information Technology • Sales • Security • Cybersecurity
Lead and mentor a data science team delivering AI-driven insights and predictive analytics for enterprise-level business applications, ensuring strategies align with operational excellence.
Top Skills: AWSAzureGCPPythonRSnowflakeSQLTableau
50 Minutes Ago
Remote or Hybrid
2 Locations
95K-140K Annually
Mid level
95K-140K Annually
Mid level
Cloud • Computer Vision • Information Technology • Sales • Security • Cybersecurity
The role involves engaging with management-level individuals to enhance security programs and assess cybersecurity maturity, while producing high-quality reports.
Top Skills: Cybersecurity
50 Minutes Ago
Remote or Hybrid
USA
170K-260K Annually
Senior level
170K-260K Annually
Senior level
Cloud • Computer Vision • Information Technology • Sales • Security • Cybersecurity
The Principal AI Data Engineer is responsible for designing and operating AI-driven data platforms, focusing on Snowflake and LLM integrations, while leading technical initiatives and mentoring teams.
Top Skills: AWSDbtGeminiGitlabGleanSalesforce AiSnowflakeTerraform

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