Senior AI Research Engineer

Skills
AWS SagemakerArtificial IntelligenceData LoadingDeep LearningHigh Performance ComputingImage ProcessingOptimising
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
  • 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

Nice to have

  • Proficient with AWS/GCP infrastructure
  • Experience in Python
  • Experience in deep learning frameworks
  • Cloud computing knowledge
  • Software engineering skills
  • Optimising model inference
  • Experience with Torch, SageMaker, CUDA
  • Understanding of Generative AI
  • Experience with deep learning, LLMs, computer vision
  • Experience in AI model training
  • Team player attitude
  • Curiosity for new AI tools and techniques
  • Experience with ML algorithms and tools

Day to day

  • Involved in training, evaluating, and engineering the training/inference infrastructure for large models for intelligence in visual domains
  • Work with and interpret petabyte-scale data
  • Build and optimise efficient systems for model research and/or apply AI techniques to create novel user experiences
  • Mentor junior team members and partake in strategic planning

Hiring process

  • Round 1 - Meet Your Recruiter
  • Round 2 - Craft Interview
  • Round 3 - Technical Task
  • Round 4 - Values Interview