Pickle Robot Company is looking to revolutionize the future of warehouse automation by advancing their perception stack and deploying robust 2D/3D vision models to production hardware to ensure their robots can handle the chaotic reality of the physical world.
Requirements
- 5+ Years of Experience in Computer Vision and Machine Learning, with a track record of shipping ML products to the physical world (Robotics, AV, or IoT).
- Expert-level Python and PyTorch skills. Working knowledge of C++ for deployment and system integration.
- Experience with 2D Vision (YOLO, MaskRCNN, Transformers) and 3D Vision (PointNet, grasp generation, multi-view geometry, camera calibration).
- You are proficient with inference optimization tools such as TensorRT, ONNX Runtime, or CUDA to maximize hardware utilization.
- You have experience curating large-scale datasets, detecting statistical bias, and automating quality assurance within the ML pipeline.
- Familiarity with Docker, AWS/GCP (S3, EC2), labeling platforms and experiment tracking tools.
Responsibilities
- Architect Multi-Modal Vision Systems: You will design and train models that fuse 2D inputs with 3D geometry to solve complex grasping and scene understanding tasks.
- Lead End-to-End Model Deployment: You will own the transition from research to reality. This includes model graph optimization, quantization (TensorRT), and runtime integration to ensure low-latency inference on our edge compute hardware (NVIDIA Orin).
- Drive Technical Excellence: As a senior member of the team, you will conduct rigorous code reviews, mentor junior engineers, and contribute to the strategic perception roadmap.
- Own the Data Strategy: You will take ownership of our existing labeled dataset and pipeline. You will identify bottlenecks, improve data quality, implement active learning strategies to systematically resolve edge cases and improve model robustness.
- Ensure Production Reliability: You will write high-performance production code (Python/C++) to seamlessly integrate perception outputs into the broader robotic control stack, prioritizing safety and system stability.
Other
- 5+ Years of Experience
- Unlimited vacation, along with all federal and state holidays
- 401K contributions of 5% your salary
- Health, dental, & vision insurance
- Travel supplies, and other items to make your working life more fun, comfortable, and productive.
