ML Engineer
Unify is seeking an ML Engineer to join their team and extend the Ivy codebase, focusing on expanding into the PyTorch Ecosystem. Responsibilities include managing ML research interns, designing graph compilers, collaborating with open-source partners, and implementing SOTA models.
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've just raised a round of venture funding (YC, Abstraction, Essence, A Capital). We’re looking for talented developers to join this ambitious endeavor, just as we’re getting started. Hop on board and lets-unify.ai!
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, Transformers, PyG, Pyro, Kornia.
We are looking for developers to join our team full-time, to extend Ivy’s codebase as we expand into the PyTorch Ecosystem and beyond. During the role, your tasks would include:
Posted June 4, 2026