JAX Internals Expert
AfterQuery · Remote
- Pay
- $75/hr
- Commitment
- contract
- Hours / week
- ~6
- Source
- afterquery
Apply on afterqueryGoes straight to the source — we never paywall the apply link.
About this role
We're sourcing ML, systems, and distributed systems engineers with deep, internals level JAX experience (e.g. custom transformations, XLA/MLIR compiler work, custom primitives, distributed sharding, custom kernel authoring) — full time professional or research experience required.
Skills & domains
- ai-training
- expert
- coding
- software-engineering