Machine Learning Software Engineer, Imbue

$140-350k

+ $10,000–$5,000,000 in equity

Python

PyTorch

Junior, Mid and Senior level

San Francisco Bay Area

Remote from US

Training AI in reasoning

Open for applications

Training AI in reasoning

21-100 employees

B2BArtificial IntelligenceDeep TechMachine LearningSaaS

Open for applications

$140-350k

+ $10,000–$5,000,000 in equity

Python

PyTorch

Junior, Mid and Senior level

San Francisco Bay Area

Remote from US

21-100 employees

B2BArtificial IntelligenceDeep TechMachine LearningSaaS

Company mission

To empower humans in an age of AI by creating powerful computing tools for people.

Job

Company

Role

Who you are

Very comfortable writing Python
Familiar with PyTorch and training deep neural networks
Excited to work on open source code
Passionate about engineering best practices
Self-directed and independent
Excellent at getting things done

What the job involves

As a remote machine learning engineer, you’ll work very closely with a senior member of our research team on cutting-edge deep learning research, infrastructure, and tooling towards the goal of creating general human-like machine intelligence
Implement a self-supervised network using contrastive and reconstruction losses
Create a library on top of PyTorch to enable efficient network architecture search
Open source internal tools
Implement networks from newly published papers
Work on tools for simple distributed parallel training of deep neural networks
Develop more realistic simulations for training our agents
Design automated methods and tools to prevent common issues with neural network training (e.g. overfitting, vanishing gradients, dead ReLUs, etc)
Create visualizations to help us deeply understand what our networks learn and why

Our take

Since the widened attention placed on generative AIs such as Chat GPT, it seems that investors have intensified their interest in general intelligence. Imbue is one of the most recent AI research and training companies to step into the limelight after it achieved unicorn status in a $200 million Series B funding round in 2023.

With competition including the likes of OpenAI and Anthropic, Imbue researches and improves the reasoning of AI models. Its focus on foundational AI means that its training has a wide range of lucrative applications. Using a full-stack approach (essentially building its tech from scratch) enables the creation of feedback loops that gradually develop even better AI models.

Freddie

Company Specialist at Welcome to the Jungle

Company

Company HQ

SoMa, San Francisco, CA

Leadership

Kanjun Qiu

(Co-Founder & CEO)

Current General Partner at Outset Capital and Sequoia Capital Scout. Previously CEO and Co-Founder at Sourceress and Member at South Park Commons.

Josh Albrecht

(CTO)

Current General Partner at Outset Capital. Previous CTO at Sourceress, Member at South Park Commons, and Public Data Lead at Addepar.

Salary benchmarks

We don't have enough data yet to provide salary benchmarks for this role.

Submit your salary to help other candidates with crowdsourced salary estimates.

Share this job

View 4 more jobs at Imbue