Lead development of multi-modal 2D/3D perception systems for warehouse robots, own dataset and data pipeline, deploy optimized models to NVIDIA Orin edge hardware, write production-grade Python/C++ code, mentor engineers, and drive model robustness via active learning and QA.
Pickle is on the hunt for a dynamic and driven Senior Machine Learning Engineer to revolutionize the future of warehouse automation. This role will focus on advancing our perception stack by refining our data engines, architecting robust 2D/3D vision models, and deploying them to production hardware to ensure our robots handle the chaotic reality of the physical world.
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.
Skills:
- 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.
- You can translate high-level product requirements into specific engineering tasks and explain technical trade-offs to non-expert stakeholders.
- Familiarity with Docker, AWS/GCP (S3, EC2), labeling platforms and experiment tracking tools.
Pay at Pickle
At Pickle Robot Company, we believe transparency builds trust. The salary range listed here is provided in accordance with Massachusetts law and reflects what we reasonably and in good faith expect to offer for this role. We often consider candidates at different levels of seniority, and final compensation will reflect the level at which a candidate is hired, along with factors like experience and location.
About Pickle Robot
Want to get in on the ground floor of a fast-growing, VC-backed robotics company? Join Pickle Robot! We build systems that companies and their teams love.
Pickle robots unload trucks. Every day, millions of trucks and containers are loaded and unloaded, often requiring manual labor—tough, dirty, dangerous, and hard to staff. Pickle automates this process using AI, machine learning, and robotics to deliver reliable products. Our Unload Systems work with teams on loading docks to make the job safer, faster, and more efficient.
Pickle provides best-in-class benefits including health, dental, & vision insurance; unlimited vacation, along with all federal and state holidays; 401K contributions of 5% your salary, travel supplies, and other items to make your working life more fun, comfortable, and productive.
Similar Jobs
Artificial Intelligence • Machine Learning • Robotics • Software • Transportation • Design • Manufacturing
Develop and integrate detection, tracking, classification, and scene-understanding ML models for on-vehicle perception. Define and implement on-vehicle tracking architecture, collaborate with model and data teams, optimize models for high-performance vehicle deployment, and work with perception input/output data.
Top Skills:
CudaPythonPyTorchSparse-BevTensorrt
Artificial Intelligence • Machine Learning • Robotics • Software • Transportation • Design • Manufacturing
Design and develop auto-labeling algorithms for autonomous driving, collaborating with ML engineers to enhance data quality and efficiency in model training.
Top Skills:
Data Management ToolsRay
Artificial Intelligence • Machine Learning • Robotics • Software • Transportation • Design • Manufacturing
Lead the development of machine learning algorithms focusing on perception for autonomous driving, collaborating across teams and mentoring ML developers.
Top Skills:
C++Deep LearningMachine LearningPython
What you need to know about the San Francisco Tech Scene
San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.
Key Facts About San Francisco Tech
- Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
- Major Tech Employers: Google, Apple, Salesforce, Meta
- Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
- Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
- Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
- Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine
