- Remote
- Full-time
- Contract
$90 - $120 per hour
... 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, https://jobeax.com/link/9JdZk1KKklJ04dJu, MLIR). - Experience with modern accelerators such as TPU (v4 to v6e), H100, B200, Trainium or ...
Australia