Software Engineer, Machine Learning

Imbue is seeking a Machine Learning Engineer to join their innovative team working on cutting-edge AI systems. The role offers a unique opportunity to work directly on creating software with human-like intelligence, focusing on deep learning research, infrastructure, and tooling.

The position involves close collaboration with senior research team members, implementing advanced papers to improve code generation, reasoning traces, and code verification. You'll be responsible for designing rigorous experiments, contributing to data generation pipelines, and developing infrastructure for fine-tuning and experimentation.

The ideal candidate should be proficient in Python, have experience with PyTorch and neural networks, and demonstrate strong engineering practices. The role requires someone who is self-directed, independent, and excellent at execution. You'll be working in an environment that values continuous learning and offers substantial resources for professional development.

Imbue's mission is to build AI systems that can reason and code, enabling AI agents to accomplish larger goals and work safely in the real world. The company trains its own foundation models optimized for reasoning and prototypes agents on top of these models. Their vision is to reinvent the personal computer concept, creating truly intelligent tools that empower users with freedom, dignity, and agency.

The compensation package is highly competitive, ranging from $140,000 to $350,000 in cash, plus significant equity potential ($10,000–$5,000,000). The role offers flexibility with both remote and in-person options in San Francisco, flexible working hours, and a dedicated budget for learning and self-improvement.

This is an exceptional opportunity for a machine learning engineer who wants to be at the forefront of AI development, working on meaningful projects that could shape the future of human-computer interaction.