Skip to content
Beacon AIBeacon AISan Carlos, CA

Software Engineer, Cloud Infrastructure

Build and operate scalable AWS cloud infrastructure powering LLM platforms, RAG systems, data pipelines, and IoT for aviation AI applications. Requires strong AWS depth, LLM/RAG production experience, Python data engineering skills, and end-to-end ownership.

135k – 260k/yr
Hybrid5+ YOEML Engineering

About the role

Key Responsibilities

Cloud Infrastructure Setup and Maintenance

  • Design, provision, and maintain AWS infrastructure using IaC tools such as AWS CDK or Terraform.
  • Build CI/CD and testing for apps, infra, and ML pipelines using GitHub Actions, CodeBuild, and CodePipeline.
  • Operate secure networking with VPCs, PrivateLink, and VPC endpoints. Manage IAM, KMS, Secrets Manager, and audit logging.

LLM Platform and Runtime

  • Stand up and operate model endpoints using AWS Bedrock and/or SageMaker; evaluate when to use ECS/EKS, Lambda, or Batch for inference jobs.
  • Build and maintain application services that call LLMs through clean APIs, with streaming, batching, and backoff strategies.
  • Implement prompt and tool execution flows with LangChain or similar, including agent tools and function calling.

RAG Data Systems and Vector Search

  • Design chunking and embedding pipelines for documents, time series, and multimedia. Orchestrate with Step Functions or Airflow.
  • Operate vector search using OpenSearch Serverless, Aurora PostgreSQL with pgvector, or Pinecone. Tune recall, latency, and cost.
  • Build and maintain knowledge bases and data syncs from S3, Aurora, DynamoDB, and external sources.

Evaluation, Observability, and Cost Governance

  • Create offline and online eval harnesses for prompts, retrievers, and chains. Track quality, latency, and regression risk.
  • Instrument model and app telemetry with CloudWatch and OpenTelemetry. Build token usage and cost dashboards with budgets and alerts.
  • Add guardrails, rate limits, fallbacks, and provider routing for resilience.

Safety, Privacy, and Compliance

  • Implement PII detection and redaction, access controls, content filters, and human-in-the-loop review where needed.
  • Use Bedrock Guardrails or policy services to enforce safety standards. Maintain audit trails for regulated environments.

Data Pipeline Construction

  • Build ingestion and processing pipelines for structured, unstructured, and multimedia data. Ensure integrity, lineage, and cataloging with Glue and Lake Formation.
  • Optimize bulk data movement and storage in S3, Glacier, and tiered storage. Use Athena for ad-hoc analysis.

IoT Deployment Management

  • Manage infrastructure that deploys to and communicates with edge devices. Support secure messaging, identity, and over-the-air updates.

Analytics and Application Support

  • Partner with product and application teams to integrate retrieval services, embeddings, and LLM chains into user-facing features.
  • Provide expert troubleshooting for cloud and ML services with an emphasis on uptime and performance.

Performance Optimization

  • Tune retrieval quality, context window use, and caching with Redis or Bedrock Knowledge Bases.
  • Optimize inference with model selection, quantization where applicable, GPU/CPU instance choices, and autoscaling strategies.

What Will Make You Successful

  • End-to-End Ownership: Drives work from design through production, including on-call and continuous improvement.
  • LLM Systems Experience: Shipped or operated LLM-powered applications in production. Familiar with RAG design, prompt versioning, and chain orchestration using LangChain or similar.
  • AWS Depth: Strong with core AWS services such as VPC, IAM, KMS, CloudWatch, S3, ECS/EKS, Lambda, Step Functions, Bedrock, and SageMaker.
  • Data Engineering Skills: Comfortable building ingestion and transformation pipelines in Python. Familiar with Glue, Athena, and event-driven patterns using EventBridge and SQS.
  • Security Mindset: Applies least privilege, secrets management, network isolation, and compliance practices appropriate to sensitive data.
  • Evaluation and Metrics: Uses quantitative evals, A/B testing, and live metrics to guide improvements.
  • Clear Communication: Explains tradeoffs and aligns partners across product, security, and application engineering.

Bonus Points

  • 4+ years working with serverless or container platforms on AWS.
  • Experience with vector databases, OpenSearch, or pgvector at scale.
  • Hands-on with Bedrock Guardrails, Knowledge Bases, or custom policy engines.
  • Familiarity with GPU workloads, Triton Inference Server, or TensorRT-LLM.
  • Experience with big data tools for large-scale processing and search.
  • Background in aviation data or other safety-critical domains.
  • DevOps or DevSecOps experience automating CI/CD for ML and app services.

Skills

AWSTerraformaws cdkLangChainPythonSageMakerbedrockopensearchpgvectorCloudWatchOpenTelemetryglueathenaeventbridgeSQS

Similar roles

ML Engineering jobs
Beacon AI

Software Engineer, Artificial Intelligence/LLM

Beacon AISan Carlos, CA

Build and ship production LLM-powered features for aviation safety and efficiency, including RAG, tool-calling, evals, guardrails, and monitoring for cost/latency/quality. Requires prior shipped LLM applications, strong production coding, and RAG depth; hybrid in San Carlos with multiple seniority levels available.

135k – 260k/yrHybrid5+ YOEML Engineering
Assembled

Software Engineer - Forecasting & Scheduling

AssembledUnited States

Builds forecasting interfaces, data pipelines, and scheduling systems for thousands of support agents using ML models. Requires Python ML libraries experience and background in ML/algorithmic teams with focus on performance optimization.

135k – 280k/yrRemoteML Engineering
Assembled

Software Engineer - Forecasting & Scheduling

AssembledSan Francisco, CA

Develops forecasting interfaces, data pipelines, and scheduling systems to predict support contact volume and optimize agent schedules for thousands, incorporating ML models and constraints like labor laws. Requires Python/ML experience and performance focus.

135k – 280k/yrHybridML Engineering
Assembled

Software Engineer - AI Agents & Platform

AssembledSan Francisco, CA

Builds autonomous AI agents and platform infrastructure for customer support, enhancing LLM performance with RAG techniques, scaling systems with Golang, and integrating STT/TTS. Requires 5+ years software engineering experience with LLMs.

135k – 280k/yrOn-site5+ YOEML Engineering
Virta Health

AI Operations Engineer

Virta HealthUnited States

Serve as technical SME driving adoption of agentic AI platform across departments. Build AI observability dashboards, govern vendor risks, integrate AI systems, and create repeatable IT workflows for safe, scalable generative AI use.

136k – 176k/yrRemote4+ YOEML Engineering