AI Research Engineer

Job Summary

To push the boundaries of AI model quality and efficiency, the full-time Research Engineer will build fine-tuning pipelines, develop benchmarking frameworks, and collaborate on hardware-aware optimization in a remote work environment.

Key Responsibilities:

• Research and implement techniques to accelerate AI inference, focusing on model quality, latency, throughput, and power consumption tradeoffs
• Collaborate with hardware, compiler, and quantization teams to ensure algorithmic improvements maximize throughput and minimize power on silicon
• Develop profiling tools and benchmarking frameworks to identify compute bottlenecks and measure model quality and efficiency metrics

Required Qualifications:

• 5+ years of experience in ML research, applied ML, or ML systems
• Strong fundamentals in Python and PyTorch
• Hands-on experience with modern AI models, including transformers and diffusion models
• Experience in fine-tuning large models and building training/evaluation pipelines
• Deep understanding of transformers, attention mechanisms, and optimization techniques