Senior Staff Engineer

About the role

Senior Staff Engineer at ShareChat. Distributed training at recommender scale. Our models are embedding-heavy —.

What you'll do

• Distributed training at recommender scale. Our models are embedding-heavy —
• ID vocabularies in the 10⁷–10⁸ range, sharded across workers and churning
• Embedding placement and sharding strategy, eviction, multi-host data-parallel
• Throughput, and holding cost per experiment down across ~35 models in daily
• Training. We run on TensorFlow today and are moving to the PyTorch ecosystem

What you need

• 8–14 yrs of experience
• Machine learning
• TensorFlow
• PyTorch
• MLOps
• Kubernetes
• Deep learning
• Any graduate