Sciforium
GPU Kernel Engineer
San Francisco, CA · Senior
Stay score
odds of building a lasting career here
Thin sponsorship signal and lottery-bound. A low-probability bet with your clock running. Prioritize cap-exempt roles and proven entry-level sponsors first.
Lottery odds assume a STEM candidate.
Personalize to your clock →Employer immigration record
from this employer's Department of Labor filings
Files H-1B transfers
Sourced from Department of Labor LCA, PERM and prevailing-wage disclosure data. Employer matching is by name, so figures may be split across an employer's legal entities. Absence of a filing means none appears in our copy of the data, not that none exists.
Community outcomes
No reports yet — be the first to help the next applicant.
About the role
- We are seeking a highly skilled GPU Kernel Engineer who is passionate about pushing the limits of performance on modern accelerators.
- You will work across the hardware-software stack, from low-level kernel development to integrating optimized ops into high-level ML frameworks used for large-scale training and inference.
- This role is ideal for someone who thrives at the intersection of GPU programming, systems engineering, and cutting-edge AI workloads, and who wants to make meaningful contributions to the efficiency and scalability of our ML platform.
Responsibilities
- Design, implement, and optimize custom GPU kernels using C++, PTX, CUDA, ROCm, Triton, and/or JAX Pallas.
- Profile and optimize end-to-end performance of ML operations, with a focus on large-scale LLM training and inference.
- Develop performance models, identify bottlenecks, and deliver kernel-level improvements that significantly accelerate AI workloads.
- Collaborate with ML researchers, distributed systems engineers, and model-serving teams to optimize compute performance across the stack.
Requirements
- Bachelor's, Master's, or PhD in Computer Science, Computer Engineering, Electrical Engineering, Applied Mathematics, or a related field.
- Hands-on experience with Triton and/or JAX Pallas for custom kernel development.
- Strong understanding of PTX, GPU ASM, and low-level GPU execution.
- Experience with AMD GPUs and ROCm optimization.
- Experience with efficient model serving frameworks (e.g., vLLM, TensorRT).
- Experience with TPUs, XLA, or similar accelerator programming environments.
Compensation
- Competitive salary and equity
Benefits
- Medical, dental, and vision insurance
- Flexible time off
- Competitive salary and equity
Company info
- ABOUT THE ROLE We are seeking a highly skilled GPU Kernel Engineer who is passionate about pushing the limits of performance on modern accelerators.
Equal opportunity
- Sciforium is an equal opportunity employer.
- All applicants will be considered for employment without attention to race, color, religion, sex, sexual orientation, gender identity, national origin, veteran or disability status.
Visa & Work Authorization
- Backed by multi-million-dollar funding and direct sponsorship from AMD with hands-on support from AMD engineers the team is scaling rapidly to build the full stack powering frontier AI models and real-time applications.
This listing is sourced directly from Sciforium's careers page and normalized into a canonical job model.