TensorFlow Expert

We're sourcing ML, systems, and distributed systems engineers with deep, internals level TensorFlow experience (e.g. custom ops and kernels, tf.function/AutoGraph internals, distributed training internals, graph level compiler work), full time professional or research experience required. Fully remote, rolling onboarding.

Responsibilities

• Complete assigned technical tasks involving TensorFlow
• Ensure code runs correctly through full runtime, including any extended, unattended runtime
• Complete tasks independently within a rolling onboarding schedule

Requirements

• Full time professional or research experience with TensorFlow
• Demonstrated internals level experience in at least one of the following: custom C++/CUDA ops and kernels, tf.function/AutoGraph tracing internals, distributed training internals (tf.distribute, collective ops), or graph level compiler work (Grappler, XLA integration)
• Access to suitable hardware (GPU enabled machine or cloud instance)

Preferred Qualifications

• Background in ML, systems engineering, or distributed systems engineering
• Contributions to TensorFlow or adjacent open source libraries (Keras, TF Extended)

Why Apply

• Work on technical tasks matched to your area of expertise
• Flexible, fully remote engagement
• Opportunity for ongoing work as new tasks arise