- Remote
- Full-time
- Contract
$90 - $120 per hour
... scale (vLLM, SGLang, TensorRT-LLM, JetStream, KV cache, paged attention, continuous batching). - Working production experience with JAX and/or PyTorch. Framework-level depth is a strong plus: custom operators, sharding and distributed training (pjit/shard_map, FSDP, Megatron, MaxText), or compiler and graph-level work (XLA/StableHLO, ...
Australia