As a Senior Engineer, you will work on open source and consulting projects, focusing on building & packaging complex projects with Python, C, C++ and possibly CUDA and Rust code for the PyPI and Conda ecosystems. You'll contribute to open-source projects, develop new features, and maintain projects with millions of users, interacting with clients from industry and academia.

As a research engineer for Learnable Planner you will support integration of new AI technologies into our autonomy/planner stack enabling our launch of fully driverless autonomous trucks, contributing towards Waabi's vision of a single AI system that learns end-to-end and in a provably safe manner as well as our revolutionary high-fidelity, closed-loop simulator, Waabi World.

The ML Training Team’s core mandate is training models as fast as possible for the company, ensuring models have 100% gpu utilization and can scale linearly from 8 gpus -> 256 gpus. Empowering MLEs by building profiling/debugging tools, setting up efficiency monitoring and integrating the trainer into the experiment management system. Optimize dataloaders / training data formats and distributed training configurations.