Remote Senior AI Research Engineer Job in California, United States Leonardo.Ai

Who you are

• Proficient with building and managing AWS/GCP infrastructure
• Experience in Python, deep learning frameworks, cloud computing, and software engineering
• Optimizing inference of models using compiling, pruning, or other techniques
• Deploying and optimizing models 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 capable of collaborating with cross-functional teams
• Curious and stays updated with new AI tools and techniques
• Experience with machine learning algorithms and tools (e.g., Pytorch, JAX), artificial intelligence, deep learning, and natural language processing

What the job involves

• Join the brightest minds in AI to unleash the world's creativity through novel tools and intelligent systems
• Train, evaluate, and engineer training/inference infrastructure for large models in visual domains
• Work with petabyte-scale data, build and optimize efficient systems for AI research, and create novel user experiences
• Mentor junior team members and play a key role in strategic planning
• Solve engineering problems to optimize model training across thousands of GPUs
• Profile and diagnose latency points in optimization and network communications
• Design high-performance distributed systems for inference and training
• Build infrastructure for complex multimodal data such as video, imagery, 3D data types, and text
• Take AI research output from prototype to production-ready code
• Collaborate with technical teams on new features, improvements, and platform stability

Application process

• Meet Your Recruiter
• Craft Interview
• Technical Task
• Values interview