Work on advanced JAX engineering tasks involving compiler internals, custom transformations, distributed sharding, XLA/MLIR integration, and high-performance machine learning systems for frontier AI research.
TensorFlow Expert
Remote (Global)
$75 – $75/hr
July 16, 2026
Job description
Role Overview
AfterQuery is seeking experienced TensorFlow experts with deep, internals-level knowledge of the framework to work on advanced machine learning and systems engineering tasks. This role is designed for professionals and researchers who have worked directly with TensorFlow internals, including custom operations, graph compilation, distributed training, and runtime optimization.
This is a fully remote contract opportunity with rolling onboarding, offering flexible work on highly technical AI and machine learning projects.
Key Responsibilities
Complete Advanced TensorFlow Tasks
- Complete assigned technical tasks involving TensorFlow.
- Develop, modify, and optimize TensorFlow-based solutions.
- Ensure implementations execute correctly through the full runtime, including extended unattended execution.
Work with TensorFlow Internals
- Build or maintain features involving:
- Custom C++/CUDA operations
- Custom TensorFlow kernels
- tf.function internals
- AutoGraph tracing
- Distributed training
- Collective operations
- Graph-level optimization
- Grappler
- XLA integration
Deliver High-Quality Implementations
- Complete technical assignments independently.
- Ensure reliability, correctness, and performance.
- Work within the rolling onboarding and task assignment process.
Required Qualifications
Full-time professional or research experience using TensorFlow.
Demonstrated internals-level expertise in one or more of:
- Custom C++/CUDA TensorFlow operations
- Custom TensorFlow kernels
- tf.function internals
- AutoGraph tracing internals
- Distributed training (tf.distribute)
- Collective operations
- Graph compiler development
- Grappler
- XLA integration
Access to suitable hardware, including:
- GPU-enabled workstation
- Cloud GPU instance
Preferred Qualifications
Background in:
- Machine Learning Engineering
- Systems Engineering
- Distributed Systems Engineering
Contributions to:
- TensorFlow
- Keras
- TensorFlow Extended (TFX)
- Related open-source machine learning projects
Compensation
- $75 USD per hour
Work Arrangement
- Remote
- Independent contractor
- Flexible schedule
- Rolling onboarding
Apply now
You will be redirected to the company's website to complete your application.
Intelligent machines powered by Artificial Intelligence computers that can learn, reason, and interact with people are no longer science fiction. Today, a self-driving vehicle powered by AI can meander through a country road at night and find its way.
Our work at NVIDIA is dedicated towards a computing model focused on visual and AI computing. For two decades, NVIDIA has pioneered visual computing, the art and science of computer graphics, with our invention of the GPU.
Who are we? We are a team of obsessive thinkers, builders, and doers from Harvard, CMU, UChicago, UPenn, GT, and more. We have published papers, own patents, built companies, and received international accolades.