Related skills
kubernetes tensorflow pytorch llms jax📋 Description
- Accelerate researchers by handling heavy ML pipelines.
- Bridge research and production: checkpoints, evaluation, APIs.
- Experiment with large DL models on thousands of GPUs.
- Design and implement ML algorithms in Python.
- Deliver prototypes that become production-grade components for Le Chat and API.
🎯 Requirements
- Master’s or PhD in Computer Science (or equivalent).
- 4+ years experience with large-scale ML codebases.
- Hands-on with PyTorch, JAX, or TensorFlow; distributed training (DeepSpeed / FSDP / SLURM / K8s).
- Experience in DL, NLP or LLMs; CUDA or data-pipeline skills a bonus.
- Strong software design: testing, code review, CI/CD.
- Self-starter, low-ego, collaborative.
🎁 Benefits
- Competitive salary and equity.
- Comprehensive healthcare for you and family.
- 401K with 6% matching.
- 18 days PTO.
- Parking or $120/mo transit stipend.
- Visa sponsorship available.
🛃 Visa sponsorship
Meet JobCopilot: Your Personal AI Job Hunter
Automatically Apply to Engineering Jobs. Just set your
preferences and Job Copilot will do the rest — finding, filtering, and applying while you focus on what matters.
Help us maintain the quality of jobs posted on Empllo!
Is this position not a remote job?
Let us know!