Senior AI Research Engineer
Skills
AWS SagemakerArtificial IntelligenceData LoadingData PipelinesDeep LearningHigh Performance ComputingOptimising
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
