Machine Learning Engineer

The Associated Press is an independent global news organization dedicated to factual reporting. Founded in 1846, AP today remains the most trusted source of fast, accurate, unbiased news in all formats and the essential provider of the technology and services vital to the news business. More thanhalf the world's population sees AP journalism every day.

Why this role matters:
The ML Engineer is a new role within the AP Engineering organization, responsible for shaping how we build and scale machine learning systems at AP, helping to lay the foundation for our machine learning capabilities. The ML Engineer has hands-on experience building and optimizing ML inference systems that run in production environments. This role will develop and tune pipelines that transform millions of photos, videos, and text documents into searchable representations using a combination of deep learning models (e.g., DistilBERT, SBERT, TransNetV2) and external multimodal APIs. The ideal candidate has experience optimizing inference at scale, orchestrating ML workloads, and working with both PyTorch and TensorFlow in a cloud environment, focusing on model performance, integration patterns, and inference efficiency.

This is an individual contributing role who will report directly to our Director of Development, Enterprise Application Services.

What you will do:

• Design, build, and scale ML-powered inference systems that process large volumes of text, image, and video data to power news-based intelligence products.

• Productionize and optimize state of the art models and inference pipelines. These models include, but are not limited to:

• • DistilBERT for Named Entity Recognition (NER) over hundreds of thousands of search queries/day

• • TransNetV2 for video shot boundary detection at scale for archival video as well as real-time
  • SBERT for embedding generation from textual descriptions
  • External multimodal APIs for image/video captioning

• Support hybrid search architectures by defining embedding/re-ranking interfaces, evaluation metrics, and inference performance requirements; partner with search/platform engineers on index configuration, sharding, and cluster tuning.

• Design and implement scalable data processing pipelines across hybrid CPU/GPU environments to handle millions of media assets.

• Partner with MLOps and platform engineering to enable the deployment and operation of ML systems reliably, contributing to:

• • Distributed inference architectures

• • Cloud-based execution (e.g., AWS EC2, Batch, Lambda, SageMaker)

• • Efficient resource utilization across workloads

• Optimize inference latency and throughput across distributed workloads using cloud-based resources (AWS EC2, Batch, Lambda, SageMaker, etc.)

• Build resilient asynchronous processing systems for large-scale workloads, ensuring:

• • Reliability (retries, fault tolerance)

• • Efficiency (caching, deduplication)

• • Observability (metrics, logging, traceability)

• Work closely with data scientists and product teams to iterate on models, improve performance, and deliver measurable impact in production.

• 8+ years of experience building production ML inference systems.

• Demonstrated ownership of deep-learning inference optimization in production (quantization, distillation, compilation, kernel/profile-level performance work) for transformer NLP and/or CV models.

• Experience with both TensorFlow (SavedModel, tf.data, XLA, TFLite) and PyTorch (TorchScript, ONNX, FastAPI/TorchServe)

• Hands-on experience optimizing inference pipelines on AWS infrastructure, ideally across different types of media assets.

• Experience with video frameworks/tools (e.g., FFmpeg), and working with large-scale frame-level inference.

• Demonstrated experience monitoring and debugging model latency, memory, and pipeline throughput.

• Experience with hybrid search architectures (BM25 + vector search + cross-encoder reranking).

• Familiarity with OpenAI APIs or other foundation model providers.

• Familiarity with open source HuggingFace LLMs.

• Experience with data pipeline and workflow orchestration tools (e.g., Airflow)

• A mission-driven, inclusive environment focused on both individual and collective success.

• Opportunities for professional development to help you reach your career goals.

• Access to tools, mentorship, and resources tailored to elevate your proficiency and contributions.

• Competitive medical, dental and vision coverage

• Retirement benefits

• Company paid life insurance

• Paid vacation and sick days

• Paid parental leave for any new parent

• Mental well-being resources