Senior AI Engineer

Cellebrite; New York (NY); Hybrid; USD 140,000 - 190,000; Full time

Job summary

Cellebrite in New York is seeking a talented developer to join our Corellium team and build scalable, cloud-native AI solutions on AWS. You will design and maintain AI workflows, ensure robust observability, and contribute to end-to-end data-driven investigations.

You’ll research machine learning algorithms, prototype new approaches, and work across a heterogeneous team to push for real-time inference and reliable deployments. This role requires US citizenship and strong Cloud/ML skills.

Qualifications

• Proficiency in Python and Typescript with API development experience.
• Hands-on AWS services experience (SageMaker, Lambda, Bedrock).
• Experience with ML libraries such as PyTorch, Pandas and NumPy.
• Ability to write clean, maintainable code and document best practices.
• Familiarity with GitHub workflows, secrets management and secure deployments.
• Experience with EKS and Kubernetes deployments.
• Experience with GenAI infrastructure, semantic caching, and LLM integration.
• Knowledge of observability tools like Datadog and log aggregation.

Responsibilities

• Design and implement features for an AWS-based AI solution including orchestration and API integrations.
• Develop and maintain CI/CD pipelines using GitHub Actions, CDK, and AWS infrastructure.
• Ensure code quality through unit tests, performance tests, and observability enhancements.
• Collaborate with cross-functional teams to integrate AI services into workflows and runtimes.
• Troubleshoot deployments and improve dashboards and monitoring tools.

Skills

Python

Typescript

API development

Cloud-native architectures

GitHub workflows

Observability

Secure deployment practices

Kubernetes (EKS)

GenAI infrastructure

LLM integration

Datadog

Tools

SageMaker

Lambda

Bedrock

PyTorch

Pandas

NumPy

Job description

Company Overview:

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:

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.
• Must be a U.S. Citizen