Senior Python ML Engineer @ Weights & Biases

Weights & Biases builds the leading ML experiment tracking platform. We're hiring a Senior Python ML Engineer to work on our SDK — a Python library used by researchers at OpenAI, NVIDIA, and thousands of other teams. You'll improve integration with PyTorch, TensorFlow, and JAX, write thorough unit tests, and maintain backward compatibility. Strong Python packaging skills and experience with ML frameworks are required. You should be comfortable reading research papers and translating them into clean library code.