CUDA Engineer

2 months ago
Full-time
Senior
Software Development
Fuse Energy

Fuse Energy

Fuse Energy is a leading UK electricity supplier prioritizing affordability, service excellence, and sustainability through renewable energy projects and global reinvestment efforts.

Renewable Electricity
11-50
Founded 2014
$78M raised

Description

  • Write and optimize custom CUDA kernels for core transformer inference operations.
  • Profile kernels to identify and eliminate bottlenecks in occupancy, memory throughput, and warp divergence.
  • Apply kernel fusion to reduce memory round-trips and launch overhead across inference pipelines.
  • Optimize memory access patterns and manage the memory hierarchy for maximum bandwidth utilization.
  • Implement quantisation-aware kernels and mixed-precision arithmetic to reduce latency and memory footprint.
  • Build and tune caching mechanisms for efficient autoregressive decoding.
  • Tune kernel launch configurations for target GPU architectures.
  • Benchmark kernels against existing baselines and drive measurable throughput and latency improvements.
  • Write tests for CUDA code to catch performance and correctness regressions.
  • Maintain internal CUDA libraries and contribute to coding standards and documentation.

Requirements

  • 4+ years writing production CUDA code, with a track record of shipping performance-critical kernels.
  • Deep understanding of GPU microarchitecture, warps, occupancy, register pressure, and memory hierarchy.
  • Strong CUDA C++ skills, including streams and asynchronous execution.
  • Hands-on experience profiling to diagnose compute-bound vs. memory-bound bottlenecks.
  • Experience with kernel fusion, memory coalescing, and avoiding warp divergence.
  • Experience writing quantised and mixed-precision kernels.
  • Solid grasp of parallel algorithm design and numerical precision tradeoffs.
  • Experience with transformer/attention-style kernels or autoregressive decoding (nice to have).
  • Experience building high-performance GPU libraries from scratch (nice to have).
  • Background in HPC or other latency-critical performance engineering (nice to have).
  • Exposure to multi-GPU or multi-node kernel-level optimisation (nice to have).
  • Comfortable reading PTX/SASS to validate kernel efficiency (nice to have).

Benefits

  • Competitive salary and an equity sign-on bonus.
  • Biannual bonus scheme.
  • Fully expensed tech to match your needs.
  • Breakfast and dinner allowance for office-based employees.

Interested in this position?

Apply directly on the company website

Apply Now

Similar Roles

Senior AI Software Engineer

Caseware 251-1K Internet Software & Services

Caseware is hiring a fully remote AI Platform Architect in Colombia to define, build, and operate scalable, secure, and trustworthy AI capabilities for its global accounting and audit software platform.

AWS AWS CDK Confluence DynamoDB GitHub GitHub Actions JIRA Microservices NestJS Python Terraform TypeScript
23 hours, 8 minutes ago

AI Ops Engineer GTM

Agiloft 51-250 Capital Markets

Agiloft is hiring an AI Ops Engineer, GTM to shape and deliver practical AI solutions across Sales, Marketing, Customer Success, and Revenue Operations, using customer and market insight to improve productivity, revenue outcomes, and customer experience.

CRM HubSpot Python Salesforce Snowflake
23 hours, 8 minutes ago

Manager, AI/ML Engineering

Lyra Health 1K-5K Health Care Providers & Services

Lyra Health is seeking an engineering leader to execute its machine learning roadmap by scaling and mentoring an AI/ML engineering team that delivers reliable, production-grade systems for mental health care.

AWS HIPAA Java Kotlin Kubernetes Machine Learning Microservices MLOps Neural Networks Prototyping Transformers
23 hours, 8 minutes ago

[Job-31694] Tech Lead AI Engineer | Agentic SDLC, Brazil

CI&T 5K-10K Internet Software & Services

A CI&T busca uma pessoa Tech Lead – AI Engineer/Orchestrator para liderar tecnicamente soluções de software em uma operação estratégica do setor financeiro, conectando arquitetura, desenvolvimento e agentes de IA em um ambiente de Agentic SDLC.

API Gateway AWS CI/CD DevSecOps Generative AI Java Microservices REST API
23 hours, 8 minutes ago

You're on a roll! Sign up now to keep applying.

Sign Up

Already have an account? Log in

Used by 14,729+ remote workers