Run any ML code, with any ML framework, on any hardware

ML Engineer intern

$35K - $60K
Job Type
Connect directly with founders of the best YC-funded startups.
Apply to role ›
Daniel Lenton
Daniel Lenton

About the role

We are looking for developers to join our team remotely for 3-12 months with flexible start dates. You will help to extend Ivy’s codebase as we expand into the PyTorch Ecosystem and beyond. We are hiring worldwide, no visa required. During the role, your tasks would include:

● Helping to design Ivy's graph compiler and transpiler, enabling automatic code conversions between frameworks

● Working alongside our open-source partners, helping to incorporate Ivy into their popular repositories

● Implementing SOTA models in Ivy, and adding these to our model hub


● Strong Python skills, with expertise in one of: PyTorch, TensorFlow, JAX

● Strong skills in recursive programming. Check out the Ivy Container class

● A passion for Machine Learning research, and for our vision to unify the ML frameworks!

About the interview

Apply Here: https://deep-ivy-ltd.breezy.hr/

Applications via YC are not monitored

a) paid open-source coding assessment, where you must complete one of the open tasks on the GitHub repository, and then create a PR to merge the completed task into master.

b) 10 recorded video responses (60 seconds each) to both high-level and technical questions about Ivy and it's place within ML.

c) 30-minute technical interview with myself, with some white-board coding exercises around some of the more technical aspects of Ivy's design, including the graph compiler and the transpiler.

d) potential offer!

About Ivy

At Ivy, we’re on a mission to unify all Machine Learning (ML) frameworks, making ML code cleaner, more flexible, and fully reusable. All Ivy functions can be executed using TensorFlow, PyTorch, MXNet, JAX and NumPy, without any change to the code.

We are in talks with developers from Google, Facebook, NVIDIA, Hugging Face, and other top software companies who would like to use Ivy in their popular open-source projects, to instantly support all frameworks. A few examples are: Ray, FastAI, HF Transformers, PyG, Pyro and Kornia.

We've just raised a round of venture funding (YC, Abstraction Capital, Essence, A.Capital). We’re looking for talented developers to join this ambitious endeavor, just as we’re getting started.

Team Size:10
Location:London, United Kingdom
Daniel Lenton
Daniel Lenton