Senior Staff / Principal Machine Learning Scientist, AI Inference & Optimization

Netskope

Santa Clara (CA)

On-site

USD 182,500 - 260,500

Full time

14 days+

Get more replies from employers

Send a job-specific resume in minutes.

Job summary

Netskope is seeking a Senior Staff Machine Learning Scientist in Santa Clara to own the inference and optimization layer for AI in agentic workflows. You will fine-tune models, push latency and throughput on real hardware, and build a runtime that executes bounded AI tasks with real customer data signals.

You will work on quantization, KV-cache optimization, and hardware acceleration, partnering with systems and backend engineers to ship end-to-end capabilities in production environments.

Qualifications

  • MS or PhD in a technical field with focus on AI/ML research.
  • 10+ years of industry experience with 4+ years in ML/AI roles.
  • Experience with model fine-tuning and inference optimization.
  • Strong Python and capability to interface with C++ for low-level work.
  • Experience with transformer models and production-grade inference.

Responsibilities

  • Build and optimize the model inference path with quantization and KV-cache optimization.
  • Fine-tune models and develop evaluation harnesses for real accuracy and latency.
  • Design and grow the bounded task execution runtime for scalable workflows.
  • Pave the path for hardware acceleration and sparsity for larger models.
  • Collaborate with systems and backend engineers to ship end-to-end capabilities.

Skills

Python
C++ interop
Fine-tuning
Quantization
Inference runtimes
Transformers
KV cache
Hardware acceleration
Agentic AI tools

Education

MS in Computer Science / ML / EE
PhD in a related field (preferred)

Tools

vLLM
ONNX Runtime
TensorRT-LLM
llama.cpp
MLX/CoreML

Job description

About Netskope

Today, there\'s more data and users outside the enterprise than inside, causing the network perimeter as we know it to dissolve. We realized a new perimeter was needed, one that is built in the cloud and follows and protects data wherever it goes, so we started Netskope to redefine Cloud, Network and Data Security.

Since 2012, we have built the market-leading cloud security company and an award-winning culture powered by hundreds of employees spread across offices in Santa Clara, St. Louis, Bangalore, London, Paris, Melbourne, Taipei, and Tokyo. Our core values are openness, honesty, and transparency, and we purposely developed our open desk layouts and large meeting spaces to support and promote partnerships, collaboration, and teamwork. From catered lunches and office celebrations to employee recognition events and social professional groups such as the Awesome Women of Netskope (AWON), we strive to keep work fun, supportive and interactive.

Positions are available at Senior Staff and above. Candidates are assessed individually and leveled according to their specific skills and background.

About the role

As a Senior Staff Machine Learning Scientist, you own the inference and optimization layer that makes AI in agentic workflows fast, efficient, and production-grade. You fine-tune and evaluate models, push latency and throughput on real hardware, and build the runtime that executes bounded AI tasks, validated against usage from Netskope’s large customer base so you optimize where the data points, not where you guess.

What’s in it for you
  • High-impact ownership. You own the model layer of a net-new product that changes the performance and economics of agentic AI.
  • Cutting-edge, unusual stack. The hard, interesting inference problems live here: quantization, KV-cache and memory management, sparsity, fine-tuning, and hardware acceleration under real-world resource constraints.
  • Real scale to build against. Netskope’s customer footprint gives you production signals most teams never see, so you deploy, validate, and iterate fast.
What you will be doing
  • Build and optimize the model inference path: quantization, KV-cache optimization, batching, and latency/memory/throughput tuning on constrained, commodity hardware.
  • Fine-tune and evaluate models for bounded tasks; build eval harnesses that gate a capability to release on real accuracy, latency, and security relevance.
  • Design and grow the task execution runtime (bounded sub-agents), pushing toward dynamic task generation and context compaction.
  • Drive hardware acceleration / sparsity and support for larger models as the platform matures.
  • Partner with the systems and backend engineers to ship capabilities end-to-end and iterate on real production signals.
Required skills and experience
  • 10+ years of overall industry experience, with 4+ years hands-on in ML/AI (model development, fine-tuning, and inference optimization).
  • Hands-on with fine-tuning (e.g.LoRA/QLoRA), quantization (GGUF/AWQ/GPTQ), and inference runtimes (vLLM/SGLang, TensorRT-LLM, ONNX Runtime, llama.cpp, or MLX/CoreML). On-device or edge inference experience is a strong plus.
  • Strong Python; comfort reaching into C++ for low-level interop is a plus.
  • Solid grasp of transformer internals and the levers that move real inference performance and cost: KV cache, attention, batching, memory footprint.
  • Fluency with agentic coding systems and genuine curiosity about agent harnesses like Claude Code, Pi, and Codex, so you should already be building with them, or itching to.
  • Clear communication: able to distill a model or infra bottleneck into an actionable concept for cross-functional teammates.
Education
  • MS in Computer Science, Machine Learning, Electrical Engineering, or equivalent technical degree required, with a focus in AI/ML research; PhD in a related field strongly preferred.

Compensation: At Netskope, salary is one component of our competitive total rewards package. The salary range for this position is as listed below. This is a national range. For purposes of complying with applicable laws, the range applies to candidates in California, Colorado, Illinois, Maryland, New York, Washington, and other states.

The successful candidate’s starting pay will also be determined based on job-related skills, experience, qualifications, location, and market conditions.

For all sales roles, the posted salary range is the On Target Earnings (OTE) range for the role, which is the sum of base salary and target commission amount at 100% goal achievement.

In addition to salary, candidates may be eligible for other forms of compensation such as participation in a bonus plan (for non-sales roles) and a stock award program. Candidates may also be eligible for a comprehensive health plan and other benefits that can be reviewed at Netskope Benefits site.

Salary Range
$182,500 — $260,500 USD

Netskope is committed to implementing equal employment opportunities for all employees and applicants for employment. Netskope does not discriminate in employment opportunities or practices based on religion, race, color, sex, marital or veteran statues, age, national origin, ancestry, physical or mental disability, medical condition, sexual orientation, gender identity/expression, genetic information, pregnancy (including childbirth, lactation and related medical conditions), or any other characteristic protected by the laws or regulations of any jurisdiction in which we operate.

Netskope respects your privacy and is committed to protecting the personal information you share with us, please refer to Netskope\'s Privacy Policy for more details.

The application window for this position is expected to close within 50 days. You may apply by filling out the below information, or visiting our Netskope Careers site.

Get your free, confidential resume review.
or drag and drop your file here.
Similar jobs

Similar jobs worth comparing

Principal / Distinguished Engineer, Machine Learning
Principal / Distinguished Engineer, Machine Learning

Netskope • Santa Clara (CA)

On-site
USD 147,000 - 300,000
Principal / Distinguished Engineer, Machine Learning
Principal / Distinguished Engineer, Machine Learning

Netskope • Santa Clara (CA)

On-site
USD 147,000 - 300,000
Comprehensive health plan
Bonus plan eligibility
Stock award program
Sr. Manager, Quality Engineering
Sr. Manager, Quality Engineering

Netskope • Santa Clara (CA)

On-site
USD 128,000 - 261,000
Competitive salary
Bonus plan and stock awards
Comprehensive health benefits
Sr. Engineering Manager, Data Security
Sr. Engineering Manager, Data Security

Netskope • Santa Clara (CA)

On-site
USD 182,500 - 260,500
Sr. Engineering Manager, Data Security
Sr. Engineering Manager, Data Security

Netskope • Santa Clara (CA)

On-site
USD 183,000 - 261,000
Senior Engineering Manager, Data Security
Senior Engineering Manager, Data Security

Netskope • Santa Clara (CA)

On-site
USD 183,000 - 261,000
Sr. Director, Business Development and Strategic Alliances (AI Ecosystem)
Sr. Director, Business Development and Strategic Alliances (AI Ecosystem)

Netskope • United States

Hybrid
USD 123,000 - 251,000
Comprehensive health plan
Bonus plan eligibility
Stock award program
Sr. Network Automation Engineer
Sr. Network Automation Engineer

Netskope • California (MO)

On-site
USD 111,000 - 226,000
IT Support Analyst
IT Support Analyst

Netskope • St. Louis (MO)

On-site
USD 42,000 - 70,000
Sr. Software Engineer, Client Backend
Sr. Software Engineer, Client Backend

Socket.dev • Santa Clara (CA)

On-site
USD 99,000 - 200,000