Senior/Staff Software Engineer - Machine Learning Platform (Inference) jobs in United States
cer-icon
Apply on Employer Site
company-logo

Snowflake · 1 week ago

Senior/Staff Software Engineer - Machine Learning Platform (Inference)

Snowflake is a company focused on empowering enterprises to achieve their full potential through innovation and collaboration. They are seeking a Senior/Staff Software Engineer for their Machine Learning Platform team to build next-generation solutions that enable customers to perform machine learning workloads efficiently on Snowflake. The role involves defining strategies, ensuring operational excellence, and collaborating with various teams to improve machine learning capabilities.

AnalyticsArtificial Intelligence (AI)Cloud Data ServicesData ManagementEnterprise SoftwareSoftware
check
Comp. & Benefits
check
H1B Sponsor Likelynote

Responsibilities

Help define and own the roadmap, working collaboratively and proactively with senior architects, PMs, and team leadership. The initiatives include platforms and tools that enable customers to do state-of-the-art machine learning on Snowflake natively
Collaboratively build and execute a vision for incorporating new advances in machine learning in ways that best achieve the team’s business objectives
Ensure operational excellence of the services and meet the commitments to our customers regarding reliability, availability, and performance
Collaborate across other ML partner teams to continuously improve ML development velocity and capabilities at Snowflake
Support team members in delivering a high level of technical quality

Qualification

Machine Learning PlatformsLLM Inference EnginesMachine Learning FrameworksInternet Serving InfrastructureTechnical Leadership

Required

Have 7+ years of industry experience designing, building, and supporting Internet serving infrastructure, machine learning platforms, machine learning services, and frameworks
Strong track record of working with machine learning systems and/or platforms
Experience in serving LLMs using inference engines like vLLM, TensorRT-LLM, TEI, SGLang, and knowing tradeoffs between them
Experience serving fine-tuned LLMs (PEFT, DPO, RL)
Experience with several of the following frameworks: SKLearn, XGBoost, PyTorch, Tensorflow, MLflow is a plus
Previous experience in building batch and real-time ML serving systems preferred
Have built a roadmap and vision around machine learning teams, and led technical decision making with help of architects and PMs and team
BS/MS/PhD in Computer Science or related majors, or equivalent experience

Benefits

Medical, dental, vision, life, and disability insurance
401(k) retirement plan
Flexible spending & health savings account
At least 12 paid holidays
Paid time off
Parental leave
Employee assistance program
Other company benefits

Company

Snowflake

company-logo
Snowflake is a cloud data platform that provides a data warehouse as a service designed for the cloud.

H1B Sponsorship

Snowflake 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 (428)
2024 (281)
2023 (154)
2022 (182)
2021 (113)
2020 (98)

Funding

Current Stage
Public Company
Total Funding
$2.03B
Key Investors
Sequoia CapitalICONIQ GrowthAltimeter Capital
2022-04-19Post Ipo Equity· $621.46M
2020-09-16IPO
2020-02-07Series G· $479M

Leadership Team

leader-logo
Sridhar Ramaswamy
Chief Executive Officer
linkedin
leader-logo
Benoit Dageville
Snowflake Co-founder, President of Product
linkedin
Company data provided by crunchbase