Senior AI Research Engineer

Skills
AWS SagemakerArtificial IntelligenceData LoadingData PipelinesDeep LearningHigh Performance ComputingOptimising
Role

What the job involves

The main requirements, responsibilities and hiring steps.

Requirements

  • 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

Day to day

  • 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

Hiring process

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