Senior AI/ML Engineer

Sorry, this job was removed at 06:11 p.m. (CST) on Thursday, Apr 23, 2026
Be an Early Applicant
Bangalore, Bengaluru Urban, Karnataka, IND
Hybrid
Artificial Intelligence • Information Technology • Machine Learning • Software • Cybersecurity • Generative AI • Data Privacy
Protect your data with Cyberhaven's reimagined data loss prevention and insider threat solution
The Role

About the role:


We’re looking for an experienced software engineer to help drive and evolve our product, focusing on high-impact innovation projects as part of our Linea AI team. You will be working on cutting-edge technology and ideas and bringing AI prototypes to production. You will work closely with Labs team to operationalize and scale AI prototypes across many customers. 

What you’ll do:

  • Design and scale distributed systems: Architect, build, and optimize highly scalable and fault-tolerant systems that process large graph datasets at enterprise scale, handling billions of events in real time from tens of thousands of endpoints with sub-second latency.

  • Solve complex scaling challenges: Tackle real-world performance and reliability problems through deep analysis, profiling, and systematic troubleshooting in high-throughput distributed environments.

  • Build with modern infrastructure: Develop and evolve a microservices-based architecture using technologies such as Python, Kubernetes, Docker, and Redis, in a continuously improving production stack.

  • Engineer secure-by-design software: Write hardened, security-first code capable of safely processing untrusted data, resisting real-world attack vectors, and operating reliably across large-scale, internet-facing systems.

  • Productionize AI systems: Partner with research and product teams to operationalize AI prototypes, turning experimental models into robust, scalable, and production-ready services.

Who you are:

  • Proven Backend Expertise: 6+ years of experience building and scaling backend systems, with strong proficiency in Python, powering production applications used at scale.

  • Full-Stack Impact: Demonstrated track record of designing and delivering full-stack applications that are actively used in real-world, high-traffic environments.

  • Generative AI & ML at Scale: Hands-on production experience with large-scale foundational models and transformer-based architectures, including deploying, monitoring, and iterating on GenAI systems.

  • Agent & Workflow Systems: At least 1 year of experience building AI agents and orchestration workflows using frameworks such as LangChain, LangGraph, Genkit, or similar technologies.

  • ML Systems & Data Pipelines: Experience delivering machine learning systems at scale, including robust model evaluation, continuous quality improvement pipelines, and data processing using columnar databases and large messaging systems (e.g., Pub/Sub, Kafka).

  • Ownership & Innovation Mindset: Comfortable working in remote, fast-paced environments, with an entrepreneurial mindset-driven to solve complex, real-world problems through innovation and breakthrough technology.

Good to have:

  • Security & Product Domain Experience: Background building products in data security, DLP, or application security environments.

  • Cloud Data & AI Workflows: Experience with GCS and BigQuery, and familiarity with LangGraph or similar frameworks for AI agent orchestration.

  • LLM Evaluation & Quality Systems: Hands-on experience designing or operating LLM evaluation pipelines, including automated testing, monitoring, and continuous quality improvement.

Joining Cyberhaven is a chance to revolutionize data security. Traditional tools fall short, but we’ve reimagined protection with AI-enabled data lineage that analyzes billions of workflows to understand data, detect risk, and stop threats. Backed by $250M from leading investors like Khosla and Redpoint, our team includes leaders who built industry-defining technologies at CrowdStrike, Palo Alto Networks, Meta, Google, and more. This role lets you shape the future of data security, alongside experts driven to help customers protect their most valuable information.

Cyberhaven is committed to creating a diverse environment and is an equal opportunity employer. All qualified applicants will receive consideration for employment without regard to race, color, religion, gender, gender identity or expression, sexual orientation, national origin, genetics, disability, age, or veteran status.

Cyberhaven Compensation & Benefits Highlights

The following summarizes recurring compensation and benefits themes identified from responses generated by popular LLMs to common candidate questions about Cyberhaven and has not been reviewed or approved by Cyberhaven.

  • Leave & Time Off Breadth Time off is described as flexible or “unlimited,” with paid holidays and sick time, alongside remote‑friendly scheduling.
  • Healthcare Strength Health coverage includes comprehensive medical, dental, and vision plans via major carriers, with mentions of mental‑health resources, wellness days, and even pet insurance in some listings.
  • Wellbeing & Lifestyle Benefits Perks include monthly phone/internet stipends and home‑office support, plus professional development opportunities such as mentorship, conferences, and employee awards.

Cyberhaven Insights

Similar Jobs

Optum Logo Optum

Machine Learning Engineer

Artificial Intelligence • Big Data • Healthtech • Information Technology • Machine Learning • Software • Analytics
In-Office
Bengaluru, Bengaluru Urban, Karnataka, IND
160000 Employees

AuxoAI Logo AuxoAI

Senior Data Scientist

Artificial Intelligence • Software • Consulting • Generative AI
In-Office
Bengaluru South, Bengaluru Urban, Karnataka, IND

Infoblox Logo Infoblox

Senior Software Engineer

Cloud • Information Technology • Security • Software
In-Office
Bangalore, Bengaluru Urban, Karnataka, IND
2100 Employees

Unisys Logo Unisys

Machine Learning Engineer

Information Technology
In-Office
Bangalore, Bengaluru Urban, Karnataka, IND
22588 Employees
50K-100K Annually
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: Mountain View, CA
237 Employees
Year Founded: 2016

What We Do

Cyberhaven was founded by computer scientists and security experts from MIT and EPFL to rethink data security. Instead of stopping threats at the perimeter, we trace the full lineage of sensitive data, where it originates, how it moves, and who uses it. This breakthrough became the foundation of our data lineage platform, now powering Cyberhaven’s leadership in data detection and response and protecting enterprises from insider threats and data loss. Our Data Detection and Response solution transforms how enterprises protect their data with a unique technology called data lineage. Cyberhaven records every event for every piece of data – every move, copy, edit, and share to fully understand how data moves throughout your company. We developed an architecture that achieves complete visibility of your data, including as it transits unmanaged cloud apps and unmanaged devices. As data moves throughout your company, from person to person and application to application, it fragments and gets combined with other data. We calculate the lineage for every piece of data starting with its origin through every step it takes. Our product allows you to define what is risky for your organization, enforce actions to protect data, and educate your workforce in real time.

Why Work With Us

Cyberhaven is a Series D unicorn in AI-driven data security, protecting 300+ customers like Motorola, Reddit, and Snowflake. We solve next-gen DLP by tracing full data lineage and enabling real-time insights. As a value-driven company with remote and hybrid work, you’ll grow alongside bold thinkers, making real impact.

Gallery

Gallery

Similar Companies Hiring

Golden Pet Brands Thumbnail
Digital Media • eCommerce • Information Technology • Marketing Tech • Pet • Retail • Social Media
El Segundo, California
178 Employees
Kepler  Thumbnail
Fintech • Software
New York, New York
6 Employees
Onshore Thumbnail
Software
US
100 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account