Software Engineer, Machine Learning

Imbue builds AI systems that reason and code, enabling AI agents to accomplish larger goals and safely work in the real world.
$140,000 - $350,000
Machine Learning
Mid-Level Software Engineer
Remote
3+ years of experience

Description For Software Engineer, Machine Learning

Imbue is seeking a Software Engineer specializing in Machine Learning to join their team. This role offers the flexibility of being fully in-person or remote, allowing you to work from anywhere or from their San Francisco, CA office.

As a machine learning engineer at Imbue, you'll collaborate closely with a senior member of the research team on cutting-edge deep learning research, infrastructure, and tooling. The ultimate goal is to create general human-like machine intelligence. Your responsibilities will include implementing papers on architecture efficiency, training speed, loss metrics, and fine-tuning methods. You'll design rigorous experiments to evaluate proposed adjustments, improve understanding of LLM fundamentals, perform hyperparameter sweeps, and contribute to the development of monitoring and logging systems for long-running experiments.

The ideal candidate should be very comfortable with Python, familiar with PyTorch and training deep neural networks, and excited about working on open-source code. You should be passionate about engineering best practices, self-directed, independent, and excellent at getting things done.

Imbue offers a compelling opportunity to work directly on creating software with human-like intelligence. They provide flexible working hours and allocate time and budget for learning and self-improvement. The company aims to rekindle the dream of the personal computer, creating truly intelligent tools that empower users and provide freedom, dignity, and agency to pursue their passions.

Compensation is highly variable, with cash compensation ranging from $140,000 to $350,000 and equity from $10,000 to $5,000,000, depending on various factors. Imbue encourages applications even if your salary requirements fall outside this range.

If you're passionate about advancing the field of AI and want to be part of a team working on groundbreaking technology, this could be an excellent opportunity for you at Imbue.

Last updated 8 months ago

Responsibilities For Software Engineer, Machine Learning

  • Implement papers on architecture efficiency, training speed, loss metrics, and fine-tuning methods
  • Design rigorous experiments to evaluate proposed adjustments for LLMs
  • Improve understanding of LLM fundamentals (chain-of-thought, reasoning, generalization, hallucination, grokking)
  • Perform hyperparameter sweeps and contribute to sweep code
  • Develop monitoring, logging, and maintenance guides for long-running experiments
  • Collaborate with data and product engineers to improve training and evaluation datasets
  • Conduct scientific research to enhance LLM model performance

Requirements For Software Engineer, Machine Learning

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

Benefits For Software Engineer, Machine Learning

Education Budget
  • Work directly on creating software with human-like intelligence
  • Flexible working hours
  • Time and budget for learning and self improvement
  • Competitive compensation package (cash + equity)

Interested in this job?

Jobs Related To Imbue Software Engineer, Machine Learning

ML Compiler Engineer, AWS Neuron, Annapurna Labs

ML Compiler Engineer position at AWS Neuron team, focusing on developing compiler technology for deep-learning workloads and contributing to cutting-edge ML infrastructure.

Software Development Engineer, Promotion Analytics and Optimization Services

Software Development Engineer role at Amazon focusing on machine learning and data analytics for promotion optimization, offering competitive salary and benefits in Vancouver.

Software Development Engineer, Predictive Targeting

Software Development Engineer role at Amazon focusing on machine learning and predictive analytics for customer targeting systems.

AI Software Engineer

AI Software Engineer position at HelpFlow to build multi-agent AI systems, requiring strong programming and AI development experience, offering remote work with potential for full-time employment.

Programmer Analyst 4 - AI Developer

Mid-level AI Developer position at DMV IT Service LLC, focusing on Document AI application maintenance and enhancement using GCP, requiring 3+ years of experience.