B

Bedrock Robotics

Builds autonomy kits for construction equipment

Onboard Infrastructure Engineer Intern - ML Inference

Applied Machine LearningConfirmed live in the last 24 hours
Internship · Summer 2027San Francisco, CA, USANo salary listedIn-person

About the Job

Requirements

  • Currently pursuing a Bachelor of Science, Master of Science, or PhD in Computer Science, Electrical Engineering, Computer Engineering, Robotics, or a related field.
  • Demonstrate proficiency in Rust or C++, with supporting experience in PyTorch or JAX.
  • Have a solid foundation in GPU architectures, CUDA, or parallel computing.
  • Understand modern systems concepts, including multithreading, operating systems and GPU scheduling, memory management, asynchronous programming, and inter-process communication.

Responsibilities

  • Integrate open-source and proprietary large language models and vision-language-action models into the onboard Rust middleware stack alongside existing perception, planning, and control pipelines.
  • Profile and optimize model execution using TensorRT, vLLM, ExecuTorch, or custom edge inference runtimes tailored for NVIDIA Jetson Thor.
  • Streamline sensor tokenization for cameras and LiDAR to feed real-time streams directly to models without latency spikes in vehicle control loops.
  • Identify and eliminate bottlenecks across memory bandwidth, compute, and inter-process communication using tools such as Nsight Systems, Nsight Compute, and eBPF.
  • Validate performance optimizations directly on heavy autonomous machinery at test sites.

Desired Qualifications

  • Practical experience deploying neural networks on constrained hardware using TensorRT, ONNX Runtime, or ExecuTorch.
  • Experience with large language model or vision-language-action optimization techniques such as key-value cache management, FP8/INT4 quantization, continuous batching, or speculative decoding.
  • Exposure to multi-modal or vision-language-action models or robotics frameworks.

About the Company

Bedrock Robotics

Builds autonomy kits for construction equipment

Bedrock Robotics builds hardware and software that turn contractors' existing heavy equipment into autonomous machines for earthmoving on construction sites. Its Bedrock Operator kit combines sensors and onboard computing to perceive the surroundings, plan excavation, and move material alongside crews and other equipment. Site managers set the work plan, while the system carries out tasks and reports progress and performance. The company develops its technology with construction partners and equips their fleets with autonomy upgrades for large infrastructure and industrial projects. Bedrock Robotics was founded in 2024 and is headquartered in San Francisco, California.

Company size
51-200
Company stage
Series C
Total funding
$950M
Headquarters
San Francisco, CA
Founded
2024

Recently Posted Jobs

Explore more opportunities at Bedrock Robotics

See more jobs