Remote Software engineering Jobs β€’ JAX

2 results

Job listings

Machine Learning Engineer, Post Training

Groq πŸ§ βš‘πŸš€

Seeking a Machine Learning Engineer to join our advanced model development team, focusing on pre-training, continued training, and post-training of models, with emphasis on draft model optimization for speculative decoding and quantization-aware training (QAT). The ideal candidate has deep experience with training methodologies, open-weight models, and performance-tuning for inference.

Research Scientist, World Models

Waabi πŸ€–πŸš—πŸ’‘
$155,000–$269,000
USD/year
US Canada Unlimited PTO

As a Research Scientist in World Models, you will develop algorithms and productionize the next generation of World Simulation models that can reason about complex, dynamic 4D environments with minimal supervision. Your work will directly power Waabi World’s ability to comprehend the 4D physical world, simulate traffic participants and generate realistic safety-critical scenarios at scale.