AI Research Engineer (Kernel & Inference Optimization)

This position is listed on behalf of a partner company, who manages all applications and next steps. Our partner is looking for an AI Research Engineer (Kernel & Inference Optimization) based in Mexico.

You will work at the intersection of AI research, systems engineering, and high-performance model inference.

Your focus will be on developing and optimizing model-serving architectures for advanced AI systems across a range of hardware environments.

You will tackle challenges involving latency, throughput, memory efficiency, and scalability, including deployment on resource-constrained mobile and edge devices.

The role combines hands-on research with low-level engineering, giving you the opportunity to develop novel inference strategies and GPU kernels.

You will work with complex architectures spanning text, image, audio, diffusion models, and vision transformers.

Your work will involve rigorous benchmarking, production testing, and iterative optimization to translate research into measurable performance improvements.

You will collaborate with cross-functional teams in a highly technical, remote environment focused on pushing the boundaries of efficient AI systems.