Remote Software engineering Jobs · JAX

Job listings

  • Set and evolve the research direction for A1’s core intelligence, including context representation, memory, reasoning, planning, and orchestration.
  • Define evaluation frameworks that measure real-world usefulness, robustness, safety, and long-term behavior.
  • Own alignment, safety, and guardrail strategy as first-class product concerns.

A1 builds a proactive smart assistant for everyday users to bring intelligence to conversations, errands, organizing, and workflows. We are a small, high-talent-density team focused on shipping high-quality work and learning at rapid speed.

$100,000–$300,000/yr

  • Build the software backbone for autonomous foundation models, including multimodal data pipelines and training workflows.
  • Implement and iterate on LLM, VLM, and VLA architectures, owning model code paths and inference runners.
  • Deliver production-grade serving tooling for low-latency operation and own systems from architecture to iteration.

We're building the next generation of ground transportation with advanced physical AI to simplify freight challenges. Our stealth team, founded by engineers who scaled autonomous driving, is developing a new vehicle platform and focuses on creating reliable real-time autonomous systems.

Global 5w PTO

  • Contribute to the entire development cycle of cutting-edge large deep learning models, from dataset preparation to deployment.
  • Collaborate with engineers and researchers to translate research into practical applications, wearing many hats.
  • Work at an exciting moment to build something from the ground up with a kind and collaborative team.

Reka builds useful multimodal AI to empower organizations, as a globally distributed foundation model startup headquartered in San Francisco Bay Area. The team, including contributors from Google DeepMind and FAIR, embraces a remote-first culture and collaborates globally.

  • Define and execute the research strategy for core AI intelligence, including reasoning, planning, memory, and context representation.
  • Lead technical decisions on model architecture, balancing proprietary solutions with state-of-the-art models.
  • Establish and oversee AI alignment, safety, and guardrail strategies to ensure responsible system behavior.

This company develops an advanced AI platform to automate complex digital workflows. They are a small, highly experienced team with a fast-moving culture focused on cutting-edge machine learning.