Member of Technical Staff (Research Engineer - LLM Systems & Performance)

Join Contextual AI in building a future where AI doesn't just answer questions, it transforms businesses. We're revolutionizing how AI Agents work by solving AI's most critical challenge: context.

Requirements

• Bachelor’s or Master’s degree in Computer Science, Electrical Engineering, or a related technical field (or equivalent practical experience)
• Strong programming skills in Python
• Experience with at least one major ML framework: PyTorch or JAX
• Solid understanding of GPU computing fundamentals (threads/warps/blocks, memory hierarchy, bandwidth vs compute, etc.)
• Familiarity with distributed training or inference concepts (e.g., model parallelism, collective communication, disaggregated serving, KV caching)
• Interest in performance engineering: profiling, kernel fusion, memory layout, and end-to-end system efficiency
• Ability to work in a fast-paced environment, communicate clearly, and collaborate closely with other engineers and researchers

Benefits

• Equity
• Benefits