Machine Learning Engineer — AI Architecture Research

Remote Remote · Remote (world)

pytorch researchfulltime

ashby

Match & cover letter

Create a profile to see your match and get a cover letter for this role.

Create profile and match

Job description

Join a remote team as a Machine Learning Engineer focused on AI architecture research. You'll design and prototype next-generation model architectures while collaborating with engineers to ensure efficient deployment.

Details

Remote work available worldwide
Competitive compensation and meaningful equity
Experience in machine learning fundamentals and deep learning required
Hands-on experience with PyTorch or JAX

The work

Research and develop new neural network architectures
Design and run architecture-level experiments
Prototype models end-to-end from research code to training-ready implementations
Collaborate with engineers to ensure architectures are deployable and efficient