AfterQuery logo

TensorFlow Expert

AfterQuery
2 hours ago
Contract
Remote
Worldwide
AI Trainer Jobs – Train AI Systems In Your Area Of Expertise

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.

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.

Commitment: 6 hours/week | Pay: $75/hr | Type: Contract

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

Required Qualifications

  • 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