Software Engineer - Model Performance
Baseten$150,000 - $250,000/year
- Full Time
- Mid-level (3 to 4 years), Senior (5 to 8 years)
Not SpecifiedCompensation
Senior (5 to 8 years)Experience Level
Full TimeJob Type
UnknownVisa
Artificial Intelligence, AI & Machine Learning, Research & DevelopmentIndustries
Requirements
Candidates should possess a Bachelor’s degree or equivalent experience, along with 8+ years of industry experience in Deep Learning frameworks such as PyTorch or TensorFlow. Strong computer science fundamentals, including algorithms and data structures, are required, as is a demonstrated passion for software engineering with excellent C++ and Python development skills and meaningful contributions to major open-source projects.
Responsibilities
The Senior Research Engineer will be responsible for developing new models and algorithms focused on Large Language Models, Natural Language Processing, and Deep Learning, designing and implementing multi-node serving architectures, optimizing multi-LoRA inference serving systems, applying sophisticated quantization techniques, implementing speculative decoding strategies, and collaborating with engineering teams to ensure seamless integration with the NVIDIA accelerated serving stack. They will also demonstrate good engineering practices and mentor other team members, contributing to all steps of the machine learning lifecycle, from conceptualization to deployment.
Skills
Generative AI
Large Language Models
Natural Language Processing
Deep Learning
Multi-node serving architectures
Distributed LLM inference
Quantization techniques (FP4/INT4, FP8)
Speculative decoding
C++
Python
PyTorch
TensorFlow
Open-source contributions
Software engineering
NVIDIA
Designs GPUs and AI computing solutions
About NVIDIA
NVIDIA designs and manufactures graphics processing units (GPUs) and system on a chip units (SoCs) for various markets, including gaming, professional visualization, data centers, and automotive. Their products include GPUs tailored for gaming and professional use, as well as platforms for artificial intelligence (AI) and high-performance computing (HPC) that cater to developers, data scientists, and IT administrators. NVIDIA generates revenue through the sale of hardware, software solutions, and cloud-based services, such as NVIDIA CloudXR and NGC, which enhance experiences in AI, machine learning, and computer vision. What sets NVIDIA apart from competitors is its strong focus on research and development, allowing it to maintain a leadership position in a competitive market. The company's goal is to drive innovation and provide advanced solutions that meet the needs of a diverse clientele, including gamers, researchers, and enterprises.
Santa Clara, CaliforniaHeadquarters
1993Year Founded
$19.5MTotal Funding
IPOCompany Stage
Automotive & Transportation, Enterprise Software, AI & Machine Learning, GamingIndustries
10,001+Employees