Software Engineer, Model Inference, DeepMind
- Hiring Organisation
- Hackajob Ltd
- Location
- South West London, London, United Kingdom
- Employment Type
- Permanent
cutting edge AI agents developed by our exceptional team of Machine Learning and Neuroscience research scientists. Your responsibilities will include everything from creating systems for agent testing using 2D and 3D games to developing test problems within physics simulators. You will create graphical visualization of results, build competitive agent … e.g., JAX, PyTorch) or low-level programming models (e.g., Pallas, CUDA, OpenCL). Experience profiling software to identify performance bottlenecks. Experience with distributed ML systems optimization and parallelism (e.g., data, model, or pipeline parallelism). Familiarity with writing performance-optimized kernels. Understanding of LLM architecture and inference performance dynamics ...