1 day ago
London, United KingdomStaff+
Responsibilities
- Optimize large-scale deep learning training workloads on NVIDIA GPU systems.
- Optimize model inference latency and pre- and post-processing on onboard systems.
- Design, train, and deploy large deep learning models using labeled and unlabeled data.
- Define cross-team technical strategies, coding paradigms, engineering best practices, and technical standards.
- Benchmark and introduce modern tools, frameworks, and architectural patterns.
- Profile models and improve model training and inference performance.
Requirements
- PhD in computer science, computer engineering, or electrical engineering, or equivalent industry experience.
- Deep knowledge of PyTorch, TensorFlow, or JAX.
- Knowledge of model training frameworks.
- In-depth knowledge of transformer architectures and methods for accelerating transformer training and inference.
- Experience with large-scale distributed model training.
- Demonstrated experience improving model training and inference speed through profiling and performance investigation.
Tech Stack
PyTorchTensorFlow
Categories
About Rivian
Rivian designs and manufactures electric vehicles, including the R1T pickup, R1S SUV, and commercial delivery vans, selling directly to consumers and fleet operators. Founded in 2009 and headquartered in Irvine, California, it operates a manufacturing plant in Normal, Illinois, and is publicly traded on NASDAQ. The company also develops in-vehicle software and electrical architecture and offers charging and service support for its owners.