- Remote
- Remote
- Full-time
- $90–$120 per hour
- 2+ years
Design, develop, and evaluate ML infrastructure and performance solutions for generative AI systems.
Compensation
Compensation: $90 - $120 per hour
Responsibilities
- Design challenging technical tasks covering accelerator kernel development, performance profiling, distributed workload debugging, and high-throughput inference serving.
- Develop complex and adversarial engineering problems that test the capabilities of advanced AI models, including tasks involving accelerator programming and optimization.
- Create accurate, technically sound reference solutions that demonstrate best practices in ML systems engineering.
- Evaluate AI-generated solutions for correctness, efficiency, scalability, and practical applicability.
- Develop detailed technical guidelines and assessment rubrics for kernel optimization, profiler interpretation, distributed systems, and inference performance.
Requirements
- Nice to have: JAX
- Nice to have: XLA
- Nice to have: TPU-based development environments
- Nice to have: custom operators
- Nice to have: sharding