Senior AI Engineer

Cellebrite; Tysons (VA); On-site; USD 120,000 - 180,000; Full time

Job summary

Cellebrite is seeking a talented developer to join the Corellium team to build and maintain scalable, cloud-native AI solutions on AWS. You will support machine learning workflows, real-time inference, and robust observability while turning forensic data into actionable insights.

You will work end-to-end from idea to prototype, supported by a multifaceted research team, in a dynamic, security-conscious environment. This role requires strong Python/Typescript skills and cloud experience.

Qualifications

• Proficiency in Python and Typescript with API development and cloud-native architectures.
• Hands-on experience with AWS services including SageMaker, Lambda and related tools.
• Experience with ML libraries such as PyTorch, Pandas, NumPy and related data processing.
• Ability to write clean, maintainable code and contribute to documentation and best practices.
• Familiarity with GitHub workflows, secrets management and secure deployments.
• Experience with Kubernetes-based deployments (EKS) and GenAI infrastructure.

Responsibilities

• Design and implement features for our AWS-based AI solution, including job orchestration, manifest generation, and API integrations.
• Develop and maintain CI/CD pipelines using GitHub Actions, CDK, and AWS infrastructure (SageMaker, Lambda, Bedrock).
• Ensure code quality and reliability through unit testing, performance testing, and observability improvements.
• Collaborate with cross-functional teams to integrate AI services into broader investigation workflows and runtime environments.
• Troubleshoot deployment issues and contribute to dashboards and monitoring tools.

Skills

Python

Typescript

API development

Cloud-native architectures

Machine learning

Data processing

GitHub Workflows

LLM integration

Observability

Tools

AWS SageMaker

AWS Lambda

AWS CDK

Bedrock

Kubernetes

EKS

Datadog

Job description

Company Overview: Cellebrite’s (Nasdaq: CLBT) mission is to enable its global customers to protect and save lives by enhancing digital investigations and intelligence gathering to accelerate justice in communities around the world. Cellebrite’s AI-powered Digital Investigation Platform enables customers to lawfully access, collect, analyze and share digital evidence in legally sanctioned investigations while preserving data privacy. Thousands of public safety organizations, intelligence agencies and businesses rely on Cellebrite’s digital forensic and investigative solutions-available via cloud, on-premises and hybrid deployments-to close faster and safeguard communities.

Company Overview: Cellebrite’s (Nasdaq: CLBT) mission is to enable its global customers to protect and save lives by enhancing digital investigations and intelligence gathering to accelerate justice in communities around the world. Cellebrite’s AI-powered Digital Investigation Platform enables customers to lawfully access, collect, analyze and share digital evidence in legally sanctioned investigations while preserving data privacy. Thousands of public safety organizations, intelligence agencies and businesses rely on Cellebrite’s digital forensic and investigative solutions-available via cloud, on-premises and hybrid deployments-to close faster and safeguard communities. To learn more, visit us at www.cellebrite.com,https://investors.cellebrite.com/investors and find us on social media @Cellebrite.

Position Overview

We are looking for a talented and driven developer to join our Corellium team and contribute to the development of scalable, cloud-native AI solutions on AWS. This role focuses on building and maintaining infrastructure and services that support advanced machine learning workflows, real-time inference, and robust observability.

You will research and build data-driven applications. Research and develop Machine Learning algorithms to turn forensic data into investigation leads, evidence, and intelligence insights. Be a part of a heterogeneous, multi-disciplinary research team. Take projects end to end, from the inception of the idea, through data and algorithm research and POC development. The work is both analytical and experimental in nature and includes prototyping of new algorithms, investigating data, and evaluating algorithm performance.

Responsibilities

• Design and implement features for our AWS-based AI solution, including job orchestration, manifest generation, and API integrations.
• Develop and maintain CI/CD pipelines using GitHub Actions, CDK, and AWS infrastructure (e.g., SageMaker, Lambda, Bedrock).
• Ensure code quality and reliability through comprehensive unit testing, performance testing, and observability enhancements.
• Collaborate with cross-functional teams to integrate AI services into broader investigation workflows and runtime environments.
• Troubleshoot deployment issues and contribute to the continuous improvement of dashboards and monitoring tools.

Requirements

• Proficiency in Python and Typescript, with experience in API development and cloud-native architectures.
• Hands-on experience with AWS services including SageMaker, Lambda, etc.
• Experience with machine learning libraries such as PyTorch, Pandas, NumPy, and related data processing tools.
• Ability to write clean, maintainable code and contribute to shared documentation and best practices.
• Familiarity with GitHub workflows, secrets management, and secure deployment practices.
• Familiarity with Amazon EKS and Kubernetes-based deployments.
• Experience with GenAI infrastructure, semantic caching, and LLM integration.
• Knowledge of observability tools like Datadog and log aggregation strategies.n empty job that is used for creating requisitions 'from scratch'
• Must be a U.S. Citizen