Remote Senior AI Research Engineer

Summary

Join Leonardo.Ai! as an AI Research Engineer and contribute to democratizing design and igniting creativity through our groundbreaking AI-powered platform. As a key member of our team, you will be involved in training, evaluating, and engineering the training/inference infrastructure for large models for intelligence in visual domains. You will also work with and interpret petabyte-scale data, build and optimize efficient systems for model research, and/or apply AI techniques to create novel user experiences.

Requirements

• Proficient with building and managing AWS/GCP infrastructure
• Experience in Python, deep learning frameworks, cloud computing, and software engineering
• Optimising inference of models from data movement, compiling, pruning, or other techniques
• Deploying and optimising models that run over several GPUs
• Previous exposure with Torch, SageMaker, and CUDA
• Knowledge of Generative AI, particularly diffusion-based image generation and VLMs
• Engineering experience with deep learning, LLMs, computer vision, AI model training, and data pipelines
• AI model and system performance tuning and optimisation
• Team player attitude capable of collaborating with cross-functional teams
• Curious and stays on top of new and changing tools and techniques in the AI space
• Experience with machine learning algorithms and tools (e.g., Pytorch, JAX), artificial intelligence, deep learning, and natural language processing

Responsibilities

• Solve engineering problems to optimise all aspects of model training from data loading to output over 1000s of GPUs
• Profiling and diagnosing points of latency and addressing in areas of low-level optimisation and network communications
• We are designing high-performance distributed systems inference and training
• We are building infrastructure for working with complex multimodal data, such as video, imagery, 3D data types, text, and beyond
• Taking AI research output from prototype to robust, scalable, production-ready code
• Collaborate with other technical teams on new features, improvements and platform stability

Benefits

• Inclusive parental leave policy that supports all parents & carers with 18 weeks paid leave
• An annual Vibe & Thrive allowance to support your wellbeing, social connection, office setup & more
• Private healthcare and 401k matching
• Flexible leave options that empower you to be a force for good, take time to recharge and supports you personally, including remote working abroad
• Support with your professional development
• Fun and engaging company events, both virtual and in-person