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.
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.
Establish clear performance targets covering response latency, token generation speed, throughput, memory footprint, and reliability.




Build and execute controlled inference benchmarks in simulated and production environments, tracking latency, throughput, memory consumption, and error rates.
Create and maintain representative datasets and simulation scenarios for evaluating model performance under real-world and resource-constrained conditions.
Identify computational and memory bottlenecks across inference pipelines and implement solutions involving batching, networking, memory management, and other system-level optimizations.
Develop custom GPU kernels and compute shaders for mobile hardware, including solutions written in Metal Shading Language (MSL).
Work with cross-functional engineering and research teams to integrate optimized inference frameworks into production and edge-device applications.
Define evaluation methodologies, document experimental results, compare performance against established benchmarks, and continuously refine optimiz

📌 AI Research Engineer (Ciudad de México)
🏢 Jobgether
📍 Ciudad de México