1 open role mentioning Geometric Machine Learning
nvidia
NVIDIA BioNeMo is seeking a Senior Software Engineer to join the cuEquivariance team. This team develops and ships production GPU kernels and software interfaces that accelerate geometric neural networks on NVIDIA GPUs, enabling researchers in fields like molecular biology, materials science, and physics to train and deploy equivariant models at scale. The role involves CUDA kernel engineering, Python library development with PyTorch and JAX, and collaboration with research teams and external framework developers. This is an opportunity to work at the intersection of GPU computing and graph-based deep learning, with your work impacting scientific communities through production pipelines.
$184k - $357k
