Machine Learning Engineer

Summary

The job is for a remote Machine Learning Engineer at Imbue, where the primary responsibility is to work on cutting-edge deep learning research and tooling towards creating general human-like machine intelligence. The role offers flexible working hours, remote work, time and budget for learning, and a competitive compensation package.

Requirements

• 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

Responsibilities

• 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

Benefits

• Flexible working hours
• Work remotely
• Time and budget for learning and self-improvement
• Compensation packages are highly variable based on a variety of factors. If your salary requirements fall outside of the stated range, we still encourage you to apply. The range for this role is $170,000–$350,000 cash, $10,000–$2,000,000 in equity