Senior AI Engineer

Socket.dev; Tysons (VA); On-site; USD 120,000 - 180,000; Full time

Job summary

Cellebrite in Tysons, VA is seeking a talented software engineer to join the Corellium team, building scalable, cloud-native AI solutions on AWS. You will design infrastructure, implement job orchestration, and integrate AI services into investigative workflows.

The role blends research and production work, delivering data-driven ML algorithms from idea to POC. Candidates should have strong Python and TypeScript skills, experience with AWS ML services, Kubernetes, and modern CI/CD practices.

Qualifications

• Proficiency in Python and Typescript with API development experience.
• Hands-on experience with AWS services (SageMaker, Lambda) and cloud-native architectures.
• Experience with ML libraries (PyTorch, Pandas, NumPy) and data processing.
• Able to write clean code and contribute to documentation and best practices.
• Familiarity with GitHub workflows, secrets management, and secure deployment.
• Experience with Kubernetes-based deployments (EKS) and GenAI infrastructure.
• Must be a U.S. Citizen.

Responsibilities

• Design and implement features for AWS-based AI solutions, incl. orchestration and API integrations.
• Develop and maintain CI/CD pipelines with GitHub Actions and CDK.
• Improve observability with dashboards and logging.
• Collaborate with cross-functional teams to integrate AI services into broader workflows.
• Troubleshoot deployment issues and ensure reliable runtimes.

Skills

Python

Typescript

API development

Cloud-native architectures

ML concepts

Tools

AWS SageMaker

AWS Lambda

AWS Bedrock

CDK

GitHub Actions

Kubernetes

Amazon EKS

PyTorch

Pandas

NumPy

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.

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

Personal Characteristics