Remote JAX Jobs · 36 Developer & Engineering Roles
Research Scientist, Optimization Systems
Aalyria · Full-time · Python, PyTorch, TensorFlow
United States1w ago
Senior Research Engineer
AssemblyAI · Full-time · JAX, XLA, Pallas
Principal Machine Learning Engineer
Bjak · Full-time · PyTorch, Spark, vLLM
SingaporeCompanies hiring
- Bjak33
- Aalyria1
- AssemblyAI1
- Cohere1
In-demand skills
- PyTorch35
- Python34
- OpenAI11
- vLLM11
- TensorRT-LLM10
- SGLang9
- Ray2
- XLA2
China
Canada
Ireland
Sweden
Germany
Portugal
Spain
Poland
Switzerland