Machine Learning Engineer - Remote, Contract or Full Time

$80K - $300K
Job Type
Any (new grads ok)
Connect directly with founders of the best YC-funded startups.
Apply to role ›
Josh Albrecht
Josh Albrecht

About the role

As a remote software engineer, you’ll work very closely with a senior member of our research team on cutting-edge deep learning infrastructure and tooling towards the goal of creating general human-like machine intelligence. We're looking for someone who's super excited to build tools (ex: experiment tracking tools, model debugging methods, automated hyperparameter optimizers, developer tooling, etc).

If interested, apply here or on Work at a Startup!

Example projects

  • 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.

You are

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


  • Work directly on creating software with human-like intelligence
  • Very generous compensation
  • Flexible working hours
  • Work remotely with a team that understands remote work
  • Focus on coding and nothing else: no expectation that you attend standup, spend time doing sprint planning, etc.
  • Work with other great programmers who care about their craft. e.g: we have deterministic formatting, zero linter errors, wide type coverage with MyPy, and even our deep learning code has tests.

About us

Imbue builds AI systems that reason and code, enabling AI agents to accomplish larger goals and safely work in the real world. We train our own foundation models optimized for reasoning and prototype agents on top of these models. By using these agents extensively, we gain insights into improving both the capabilities of the underlying models and the interaction design for agents.

We aim to rekindle the dream of the personal computer, where computers become truly intelligent tools that empower us, giving us freedom, dignity, and agency to pursue the things we love.

About Imbue (formerly Generally Intelligent)

Generally Intelligent is an AI research company. We help machines learn to understand the world the way humans do. Our mission is to build human-like general machine intelligence and make it safely accessible in order to foster a more abundant, equitable, and creative human society.

We take a first-principles approach to understanding the fundamentals of learning, starting with simple self-supervised networks solving early evolutionary problems, and increasing complexity incrementally.

Imbue (formerly Generally Intelligent)
Team Size:15
Location:San Francisco
Josh Albrecht
Josh Albrecht
Kanjun Qiu
Kanjun Qiu