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