Would you like to enable ML use cases like autonomous driving scene understanding and automated mapping at Zoox This role works across all ML teams within Zoox Perception Behavior ML Simulation Data Science Collision Avoidance as well as with our Advanced Hardware Engineering group specifying our next generation of autonomous hardware. You will significantly push the boundaries of how ML is practiced within Zoox.
We build and operate the base layer of ML tools deep learning frameworks inference libraries and ML infrastructure used by our applied research teams for in and offvehicle ML use cases. We coordinate across all of Zoox to make sure that the needs of both the vehicle and ML teams are met. You will play a crucial role in reducing the time it takes from ideation to productionization of cuttingedge AI innovation. This team has a lot of growth opportunities as we expand our robotaxi deployments and venture into new ML domains. If you want to learn more about our stack behind autonomous driving please look here.
In this role you will:
Design develop and deploy custom and offtheshelf ML libraries and toolings to improve ML development training deployment and onvehicle model inference latency.
Build tooling and establish development best practices to manage and upgrade foundational libraries i.e. Nvidia driver PyTorch TensorRT etc. improve ML developer experience and expedite debugging efforts.
Collaborate closely with crossfunctional teams including applied ML research highperformance compute advanced hardware engineering and data science to define requirements and align on architectural decisions.
Qualifications
6 years of experience
Proficient in Python or C
Familiarity with any of the training frameworks and libraries like PyTorch Lightning Hugging Face Ray JAX etc.
Familiarity with any of the GPUaccelerated inference on Nvidia hardware like CUDA TensorRT and/or XLA
Bonus Qualifications
Familiarity with Bazel and/or the C linker
Compensation
There are three major components to compensation for this position: salary Amazon Restricted Stock Units (RSUs) and Zoox Stock Appreciation Rights. The salary range for this position is $205000 $260000. A signon bonus may be offered as part of the compensation package. Compensation will vary based on geographic location and level. Leveling as well as positioning within a level is determined by a range of factors including but not limited to a candidates relevant years of experience domain knowledge and interview performance. The salary range listed in this posting is representative of the range of levels Zoox is considering for this position.
Zoox also offers a comprehensive package of benefits including paid time off (e.g. sick leave vacation bereavement) unpaid time off Zoox Stock Appreciation Rights Amazon RSUs health insurance longterm care insurance longterm and shortterm disability insurance and life insurance.
A Final Note:
About Zoox
Zoox is developing the first groundup fully autonomous vehicle fleet and the supporting ecosystem required to bring this technology to market. Sitting at the intersection of robotics machine learning and design Zoox aims to provide the next generation of mobilityasaservice in urban environments. Were looking for top talent that shares our passion and wants to be part of a fastmoving and highly executionoriented team.
Disclaimer: Drjobpro.com is only a platform that connects job seekers and employers. Applicants are advised to conduct their own independent research into the credentials of the prospective employer.We always make certain that our clients do not endorse any request for money payments, thus we advise against sharing any personal or bank-related information with any third party. If you suspect fraud or malpractice, please contact us via contact us page.