JAX jobs
2 open roles for jax, updated continuously as employers post them.
Machine Learning Engineer
A remote Machine Learning Engineer role at Anlatan, training and fine-tuning LLMs and image models for consumer AI products. Suits an ML engineer with PyTorch experience and a background training NLP or image models.
Senior Applied ML Engineer
Senior applied ML engineer role developing learning systems for autonomous AI agents on CoreWeave's CRAFT team in Bellevue or Sunnyvale. The role suits machine-learning engineers with strong model-training experience, Python, PyTorch or JAX, and interest in LLM post-training research.