FieldAI is transforming how robots interact with the real world. We are seeking an accomplished Staff AI Software Engineer to drive the optimization, integration, and deployment of our ML models on real robotic platforms, ensuring robust on-robot behavior in complex environments.
Responsibilities:
- Convert and optimize 2D/3D CNNs and Transformer-based models (PyTorch/TensorFlow → ONNX → TensorRT/Triton) for real-time inference on Jetson/Orin platforms
- Apply model compression techniques—quantization, pruning, distillation, weight sharing—to meet strict constraints on latency, memory, bandwidth, and power
- Develop custom TensorRT plugins and CUDA kernels for performance-critical components
- Integrate optimized models into the broader robotic system using ROS nodes and interfaces
- Build benchmarks, profile and debug end-to-end inference pipelines, and validate performance in real-world robotic scenarios
- Collaborate closely with AI researchers, robotics engineers, and hardware teams to translate cutting-edge research into robust, deployable edge solutions
- Ensure the reliability, robustness, and stability of deployed models operating continuously in challenging, resource-constrained environments