Gradient

AI Research Engineer

Gradient

full-time

Posted on:

Location Type: Remote

Location: Anywhere in the World

Visit company website

Explore more

✨ AI Apply
Apply

Tech Stack

Distributed SystemsGoPythonPyTorchRustTensorflow

About the role

  • Design and implement protocols for AI training and inference (e.g. federated learning, distributed fine-tuning, secure model serving)
  • Build scalable systems to support AI workloads across heterogeneous networks and devices
  • Contribute to architectural decisions, performance optimization, and reliability engineering
  • Collaborate with researchers and system engineers to translate novel ideas into real-world systems
  • Stay current with the latest advancements in AI, distributed infrastructure, and cryptographic mechanisms

Requirements

  • Strong programming skills in Python, Rust, or Go
  • Experience with AI/ML model training and serving pipelines (e.g. PyTorch, JAX, TensorFlow)
  • Solid foundation in distributed systems, networking, or applied cryptography
  • Familiarity with concepts like federated learning, secure aggregation, or verifiable intelligence
  • Bonus: experience working with blockchain protocols, zkML, or decentralized networks
Benefits
  • Flexible work arrangements
  • Professional development opportunities
Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills & Tools
PythonRustGoAI/ML model trainingAI/ML model servingPyTorchJAXTensorFlowdistributed systemsapplied cryptography
Soft Skills
collaborationarchitectural decision-makingperformance optimizationreliability engineering