{"id":3876498,"url":"https://alion.io/job/commonai-gpu-kernel-and-performance-engineer","title":"GPU Kernel and Performance Engineer","company":{"id":38984,"name":"CommonAI","domain":"commonai.org","url":"https://alion.io/company/commonai","size_band":"51-200","is_staffing_agency":false,"employer_type":"direct","is_intermediary":false,"listed_via":null,"ats_vendor":"Workable","truth_index":null},"role":"AI/ML","role_family":"AI/ML","seniority":null,"employment_type":"full_time","work_mode":"on_site","remote_scope":null,"remote_scope_basis":null,"remote_working_hours":null,"hiring_geo_confidence":"structured","locations":["Cambridge, United Kingdom"],"countries":["GB"],"hiring_countries":[],"hiring_countries_total":0,"salary":null,"salary_estimate":null,"experience_years_min":null,"visa_sponsorship":false,"relocation_package":false,"has_equity":false,"technologies":[{"name":"Apache TVM","optional":false},{"name":"C++","optional":false},{"name":"CUDA","optional":false},{"name":"CUTLASS","optional":false},{"name":"HPC","optional":false},{"name":"JAX","optional":false},{"name":"LLM","optional":false},{"name":"Mixture of Experts","optional":false},{"name":"MLIR","optional":false},{"name":"NCCL","optional":false},{"name":"NVLink","optional":false},{"name":"Python","optional":false},{"name":"PyTorch","optional":false},{"name":"TPU","optional":false},{"name":"Triton","optional":false},{"name":"vLLM","optional":false},{"name":"XLA","optional":false}],"status":"live","first_seen_at":"2026-11-06T16:32:05Z","employer_posted_date":"2026-11-06","last_verified_at":"2026-11-07T18:00:34Z","board_verified":true,"closed_at":null,"days_open":1,"trust":{"level":"ok","repost_count":null,"flags":[],"days_open":1},"description":"CommonAI CIC is a non-profit membership organisation, founded on a belief in collaborative engineering for the safe and responsible development of foundational AI technologies. A place where AI startups, enterprises large and small, public sector bodies and academia can share resources and knowledge, to codevelop and grow businesses, fast.\nWe are led by experienced founders, investors and engineers who believe that collaborative engineering drives faster AI innovation and are backed by a mix of UK Government and private funding in order to design, build and deploy innovative AI systems.\nThe Opportunity\n We are seeking a GPU Kernel and Performance Engineer to make our research models run fast on real hardware, primarily NVIDIA GPUs. Our research team works on reducing the complexity of large language models while retaining their performance. Some of this work informs our analyses of the trade-offs between training and inference costs when scaling inference workloads. The work runs on the UK's national AI supercomputers and on our own compute nodes.\nYou will join a small research team and turn new model designs into efficient GPU code, then measure how much time and memory traffic they cost on real hardware. Our training runs on NVIDIA GPUs, and we also expect to work with other AI accelerators, so there is scope to learn new chips and programming models as the work develops. We care more about how well you understand hardware and memory than about ML research credentials.\nWhat You'll Do\nWrite and optimise CUDA and Triton kernels on NVIDIA GPUs for sparse and structured model components, including block-sparse matrix operations and attention variants.\nAdapt and extend open-source kernels, such as FlashAttention-style and block-sparse attention, to new model designs.\nProfile training and inference to find where time and memory traffic go, using tools such as NVIDIA Nsight Systems and Nsight Compute.\nOptimise for the GH200's memory hierarchy on Isambard-AI and for our own GPU nodes.\nBuild tools that measure data movement per generated token, and check analytic cost models against hardware measurements.\nTest every kernel against reference implementations before it runs at scale.\nWork with our training infrastructure engineer to raise training throughput, and with researchers to make new designs fast enough to test.\nEvaluate how our models map onto other AI accelerators, such as TPUs and dataflow, wafer-scale and compute-in-memory architectures, and port key components to them where it is useful.\nContribute improvements back to open-source projects where possible.\nRequirements\nProven experience writing and optimising GPU kernels in CUDA or Triton, ideally on recent NVIDIA architectures.\nA strong understanding of GPU architecture and memory hierarchies (e.g. shared memory, HBM bandwidth, memory coalescing, occupancy).\nHands-on experience profiling and optimising performance with tools such as NVIDIA Nsight Compute or Nsight Systems.\nStrong Python and C/C++.\nA rigorous approach to testing numerical code (e.g. comparison against reference implementations, mixed-precision error).\nOpenness to learning new hardware architectures and programming models, often from limited documentation.\nA strong enthusiasm for using LLMs, coding assistants, and agents as core tools in your own software development process.\nWe also value:\nExperience with attention kernels (e.g. FlashAttention) or sparse kernels (e.g. block sparsity, mixture-of-experts routing).\nExperience with NVIDIA Hopper features (e.g. TMA, warpgroup matrix instructions) or libraries such as CUTLASS and CuTe.\nExperience with multi-GPU communication (e.g. NCCL, NVLink).\nAn understanding of LLM inference performance (e.g. KV caching, memory-bound decoding, quantisation).\nExperience with other AI accelerators (e.g. AMD GPUs, TPUs with JAX and Pallas, or dataflow, wafer-scale and compute-in-memory architectures).\nExperience with compiler stacks for AI hardware (e.g. MLIR, XLA, TVM).\nA background in computer architecture, HPC, graphics or game engine programming.\nExperience with hardware/software co-design, or with hardware performance modelling and simulation.\nContributions to performance-critical open-source projects (e.g. PyTorch, Triton, JAX, vLLM).\nBenefits\nA collaborative and supportive work environment\nThe opportunity to have a high impact in a growing organisation\nCompetitive salary package and pension\nProfessional development opportunities\nNetworking opportunities with influential people from across the tech sector, financial services, and academia\nA vibrant office environment located a few minutes' walk away from Cambridge train station\nCommonAI CIC is an equal opportunity employer and is committed to creating an inclusive and diverse workplace.","description_format":"text","description_chars":4778,"description_truncated":false,"requirements":{"experience_years_min":null,"management_years_min":null,"team_size_min":null,"manages_managers":false,"education":{"level":"bachelor","optional":false},"security_clearance":false,"languages":[]},"benefits":["Professional development"],"hiring_locations":[],"hiring_excludes":[],"relocation_offered":false,"industries":["AI Compute & Inference","High-Performance Computing"],"lifecycle":[{"event":"open","at":"2026-11-06T16:32:05Z"}],"visa":[],"liveness":{"score":90,"band":"hot","label":"Hiring now","p_open":1,"p_active":0.903,"p_room":1,"age_days":0,"expected_fill_days":56,"reasons":["conf:4","velocity","win:early"],"computed_at":"2026-11-07T05:47:08Z"},"pay":null,"html_url":"https://alion.io/job/commonai-gpu-kernel-and-performance-engineer","json_url":"https://alion.io/job/commonai-gpu-kernel-and-performance-engineer.json","meta":{"generated_at":"2026-11-07T19:29:00Z","cache_seconds":300,"methodology":"https://alion.io/methodology","terms":"https://alion.io/terms","contact":"https://alion.io/contact","api":"https://alion.io/developers","about":"Alion is a live layer of people, companies and AI agents: who they are, whether they are real and active right now, what they do and how to work with them, readable by people and by agents and paid per call.","catalog":"https://alion.io/catalog.json","usage":{"tier":"crawler_verified","counted_by":"address","units_charged":1,"used_today":30285,"day_limit":null,"remaining_today":null,"minute_limit":300,"resets_at":"2026-11-08T00:00:00Z"}},"offers":[{"id":"company.slices","title":"One company in depth, by slice","status":"live","price":{"credits":0.02,"usd":0.002,"plus_per_slice":{"credits":0.05,"usd":0.005}},"unit":"per company, plus each slice with data","note":"the employer in depth","call":{"mcp_tool":"get_company","arguments":{"id":38984},"rest":"https://alion.io/mcp/rest/get_company?id=38984"},"human":"https://alion.io/catalog?offer=company.slices&for=job%2Fcommonai-gpu-kernel-and-performance-engineer"},{"id":"market.stats","title":"A market slice: pay, demand and time to fill","status":"live","price":{"credits":1,"usd":0.1},"unit":"per slice","note":"pay, demand and time to fill for this role and place","call":{"mcp_tool":"market_stats"},"human":"https://alion.io/catalog?offer=market.stats&for=job%2Fcommonai-gpu-kernel-and-performance-engineer"},{"id":"job.search","title":"Open jobs by role, technology, place, pay and visa","status":"live","price":{"credits":0.02,"usd":0.002},"unit":"per posting in a list","note":"similar open postings","call":{"mcp_tool":"search_jobs"},"human":"https://alion.io/catalog?offer=job.search&for=job%2Fcommonai-gpu-kernel-and-performance-engineer"},{"id":"company.verify","title":"Is this company real and active right now","status":"pilot","price":null,"unit":"per company","request":{"url":"https://alion.io/catalog/request","method":"POST","body":"{\"offer\": \"company.verify\", \"for\": \"job/commonai-gpu-kernel-and-performance-engineer\", \"note\": \"what you need it for\"}"},"human":"https://alion.io/catalog?offer=company.verify&for=job%2Fcommonai-gpu-kernel-and-performance-engineer"}],"agent":{"mcp":"https://alion.io/mcp/jobs","mcp_open":"https://alion.io/mcp/open","next":[{"tool":"get_job","arguments":{"id":3876498}},{"tool":"search_jobs","arguments":{"company":"commonai.org","detail":"compact"}},{"tool":"match_people","arguments":{"job_id":3876498},"note":"people open to offers who fit this vacancy"}],"note":"The same data through MCP tools: https://alion.io/mcp/open needs no sign-in (compact searches, postings, people); https://alion.io/mcp with an Alion account has everything. Docs: https://alion.io/developers"}}