XLA Jobs in New York
Search jobs in
1 - 7 of 7
Search Results - XLA Jobs in New York
Fintal PartnersNew York
Solid grounding in computer architecture
/n
• C++ and Python
/n
/n
/n
Bonus points for
/n
/n
• The JAX ecosystem (XLA, Flax and similar)
/n
• GPU tooling — Triton, CUB, cuDNN, cuBLAS
/n
• Linux systems programming
/n
• Large-scale distributed...
EPAM SystemsNew York
technical liaison partnering with ML researchers, core framework architects (JAX, PyTorch), and compiler engineering teams (XLA, MLIR) to eliminate systemic bottlenecks and shape future hardware/software requirements Architect foundational infrastructure...
24-MAGNew Yorkcareers-page.com
to kernel or accelerator libraries such as cuBLAS, cuDNN, Triton community kernels, or JAX/XLA custom calls are advantageous
• Strong written communication and ability to provide precise technical feedback
Engagement Details
• Part-time independent...
Fintal PartnersNew York
high-performance deep learning systems in any domain (robotics, recommendations, audio, video, biology, and so on)
Nice to Have
• Experience with Triton, CUB, cuDNN, or cuBLAS
• JAX ecosystem experience (XLA, Flax)
• Linux systems programming
• Large...
ESTÉE LAUDER COMPANIESNew Yorkfashionjobs.com
interfaces, production controls, environment readiness, capacity, disaster recovery, and operational documentation.
• Publish meaningful service reporting and insights on SLA/XLA performance, backlog, aging, availability, defects, demand, recurring issues...
Fintal PartnersNew York
Cores, cooperative groups, graphs, warp-level intrinsics
• Deep familiarity with the internals of frameworks such as PyTorch, JAX or TensorFlow
• Solid grounding in computer architecture
• C++ and Python
Bonus points for
• The JAX ecosystem (XLA, Flax...
Fintal PartnersNew York
approach to solving complex technical problems
Additional experience valued
• Experience with the JAX ecosystem, including XLA and Flax
• GPU or accelerator programming using CUDA, Triton or Pallas
• Large-scale distributed training
• Expertise in deep...
XLA jobs – More cities:
Don’t miss out on new job openings!
Create a job alert for: XLA, New York
It's free, and you can cancel email updates at any time