HimalayasHimalayas logo
Wallarm Inc.WI

Senior Machine Learning Engineer

Wallarm Inc. provides an AI-powered API security platform offering real-time threat detection and protection for web applications, microservices, and APIs across cloud-native environments.

Wallarm Inc.

Employee count: 51-200

Stay safe on Himalayas

Never send money to companies. Jobs on Himalayas will never require payment from applicants.

Since 2016, Wallarm has been on a mission to secure the internet's critical infrastructure: APIs. Today, we are the trusted choice for over 200 of the world's most innovative companies, from high-growth startups to Fortune 500 and Nasdaq leaders. Our unified platform provides full-lifecycle API security — helping teams discover their attack surface, protect against modern threats, and respond to incidents in real-time. As a graduate of Y Combinator and fueled by a recent $55M Series C, we are scaling our global, remote-first team of 150+ innovators to solve the next generation of security challenges.

We're building ML-powered detection systems that protect APIs from automated abuse credential stuffing, scraping, enumeration, and attack patterns that evolve daily. This is a greenfield effort: we have the data and the ideas, but the ML infrastructure, pipelines, and models need to be built from scratch.

You'll be the first dedicated ML engineer on the team, working closely with engineers, security researchers and DevOps. This is a senior IC role with a clear path to technical leadership - we plan to grow the ML function around this hire.

What You'll Do

  • Build the ML stack from the ground up - Design and implement the data pipelines, feature extraction, model training, and serving infrastructure needed for production-grade anomaly detection.

  • Detecting anomalies in API traffic - Your first major outcome: build a system that identifies malicious behavioral patterns across client sessions with high precision and recall, trained per-client.

  • Own the full lifecycle - From raw data exploration and feature engineering through model development, evaluation, deployment, and continuous monitoring. No handoffs to a separate "productionization" team.

  • Design experiments and metrics - Build offline evaluations, define detection-quality metrics, and monitor for false positives, drift, and adversarial adaptation.

  • Work with text and structured behavioral data - Extract signals from API sessions, request sequences, payloads, and traffic metadata using NLP and statistical techniques.

  • Leverage LLMs where they add value - Explore embedding-based models and LLM-augmented approaches for signal enrichment, classification, and explainability.

  • Shape the technical direction - Document findings, present to cross-functional teams, and help define the ML roadmap as the team grows.

Requirements

What We're Looking For

Required

  • 5+ years in Applied ML or ML Engineering with production deployment experience (not research-only backgrounds).

  • Strong NLP / text data experience - hands-on work with text classification, pattern extraction, tokenization, embeddings, or similar. This is the core of the work.

  • Proficiency in Python and production-grade systems (APIs, data pipelines, model serving).

  • Solid data engineering skills - experience building ETL/data pipelines, working with batch and streaming data, and understanding the full ML data lifecycle (DAGs, data versioning, feature stores).

  • Deep hands-on experience across ML fundamentals: classification, anomaly detection, clustering, statistical methods - and the judgment to choose the right approach for a given problem.

  • Comfort with imperfect data - noisy labels, class imbalance, evolving distributions - and practical strategies for labeling, evaluation, and shipping reliable models.

  • End-to-end ownership mindset - ability to take a problem from raw data to production deployment, working with DevOps to stand up the necessary infrastructure.

  • Strong experimentation skills: prototype fast, design rigorous evaluations, measure outcomes, reason about trade-offs (cost, quality, latency).

    Strongly Preferred

  • Experience in domains where adversaries actively adapt to detection (fraud, bot mitigation, abuse prevention, spam). The ML mindset of handling concept drift and adversarial evasion matters more than specific domain knowledge.

  • Familiarity with ML lifecycle tooling: experiment tracking (MLflow, W&B), model versioning (DVC), weak-supervision tools (Snorkel, cleanlab), drift monitoring.

  • Experience with big data / streaming stacks (Spark, Kafka, BigQuery) or cloud ML platforms (AWS SageMaker, GCP Vertex).

  • Background in security research or threat intelligence (not required - domain context can be learned).

    Who Thrives Here

  • You're a full-stack ML engineer - equally comfortable building a data pipeline and tuning a model, designing an experiment and deploying it to production.

  • You've built from scratch before - you know what it takes to go from "we have data and ideas" to "we have a working detection system."

  • You're energized by ambiguity and ownership - this isn't a well-scoped ticket queue, it's an open problem space where you define the path.

  • You're ready to grow into leadership - mentoring engineers, shaping technical strategy, and owning the ML roadmap as the team scales around you.

  • You leverage modern tools (AI-assisted development, LLM-augmented workflows) to move faster without cutting corners.

About the job

Apply before

Posted on

Job type

Full Time

Experience level

Experience

5 years minimum

Location requirements

Open to candidates from all countries.

Hiring timezones

Worldwide

About Wallarm Inc.

Learn more about Wallarm Inc. and their company culture.

View company profile

Wallarm Inc. delivers advanced API security solutions designed to protect modern web applications, microservices, and APIs operating in cloud-native environments. The company's platform provides comprehensive, real-time protection against a wide spectrum of cyber threats, including the OWASP Top 10, zero-day attacks, account takeover (ATO), malicious bots, and application abuse. Wallarm's core technology leverages artificial intelligence and machine learning to automatically discover all API endpoints, assess risks, detect anomalous behavior, and block malicious traffic with high accuracy and minimal false positives. This AI-driven approach enables the platform to generate application-specific security rules dynamically, adapting to evolving threat landscapes and application changes without requiring manual intervention.

Founded with a vision to address the escalating security challenges posed by the proliferation of APIs, Wallarm offers a unified platform that integrates API discovery, real-time threat prevention, and automated security testing. This allows security and DevOps teams to embed security throughout the application lifecycle, from development to production. The platform supports diverse deployment options, including inline, out-of-band, and as a cloud-native WAAP (Web Application and API Protection) service, catering to various infrastructure needs across public, private, and hybrid cloud environments. Wallarm is trusted by numerous enterprises, including Fortune 500 companies and high-growth startups, to secure their critical digital assets and ensure the integrity of their API-driven services. The company emphasizes continuous innovation in API security, contributing to the cybersecurity community through research and open-source projects, and aims to provide the fastest, easiest, and most effective way to stop API attacks.

Employee benefits

Learn about the employee benefits and perks provided at Wallarm Inc..

View benefits

Flexible working hours

Supports work-life balance.

Paid medicine

Allocated budget including psychologists.

Workplace equipment budget

Allocated budget for workplace equipment.

Competitive salary and bonuses

Offered as part of the compensation package.

View Wallarm Inc.'s employee benefits
Claim this profileWallarm Inc. logoWI

Wallarm Inc.

View company profile

Similar remote jobs

Here are other jobs you might want to apply for.

View all remote jobs

Remote companies like Wallarm Inc.

Find your next opportunity by exploring profiles of companies that are similar to Wallarm Inc.. Compare culture, benefits, and job openings on Himalayas.

View all companies

Find your dream job

Sign up now and join over 100,000 remote workers who receive personalized job alerts, curated job matches, and more for free!

Sign up
Himalayas profile for an example user named Frankie Sullivan