Member of Technical Staff - ML Research Engineer, Performance Optimization jobs in United States
cer-icon
Apply on Employer Site
company-logo

Liquid AI · 6 hours ago

Member of Technical Staff - ML Research Engineer, Performance Optimization

positionSan Francisco, CA
timeFull-time
remoteHybrid
seniorityMid Level

Liquid AI is a foundation model company spun out of MIT, focused on building capable and efficient general-purpose AI systems. The role of a Member of Technical Staff - ML Research Engineer involves writing high-performance GPU kernels, optimizing architectures, and implementing cutting-edge techniques in machine learning systems.

Artificial Intelligence (AI)Generative AIInformation TechnologyMachine Learning
check
H1B Sponsor Likelynote

Responsibilities

Write high-performance GPU kernels for inference workloads
Optimize alternative architectures used at Liquid across all model parameter sizes
Implement the latest techniques and ideas from research into low-level GPU kernels
Continuously monitor, profile, and improve the performance of our inference pipelines

Qualification

GPU kernel optimizationCUDAC/C++PyTorchPerformance profilingCollaborationProblem-solving

Required

You have experience writing high-performance, custom GPU kernels for training or inference
You have an understanding of low-level profiling tools and how to tune kernels with such tools
You have experience integrating GPU kernels into frameworks like PyTorch, bridging the gap between high-level models and low-level hardware performance
You have a solid understanding of memory hierarchy and have optimized for compute and memory-bound workloads
You have implemented fine-grain optimizations for a target hardware, e.g. targeting tensor cores

Preferred

CUDA
CUTLASS
C/C++
PyTorch/Triton

Company

Liquid AI

twittertwittertwitter
company-logo
Build efficient general-purpose AI at every scale.
dateFounded in 2023
location
Cambridge, Massachusetts, USA
people-size51-200 employees
web-link
http://liquid.ai

H1B Sponsorship

Liquid AI has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (2)

Funding

Current Stage
Growth Stage
Total Funding
$293.1M
Key Investors
AMD VenturesOSS Capital L.P.
2024-12-13Series A· $250M
2023-12-01Seed· $37.5M
2023-05-05Seed· $5.6M

Leadership Team

leader-logo
Ramin Hasani
Co-founder and CEO
linkedin
leader-logo
Mathias Lechner
Co-founder and CTO

Recent News

IEEE Spectrum
MIT’s AI Robotics Lab Director Is Building People-Centered Robots
2025-12-06
Digital Commerce 360
Shopify partners with Liquid AI on product search, recommendations
2025-11-15
Liquid AI
Liquid AI Announces Multi‑Year Partnership with Shopify to Bring Sub‑20ms Foundation Models to Core Commerce Experiences
2025-11-14
Company data provided by crunchbase