1 to 25 of 41 Remote JAX Jobs

related field, with at least 8 years of experience. Strong programming skills in Python and proficiency with Deep Learning frameworks (e.g., PyTorch, JAX, TensorFlow). Demonstrated hands-on experience implementing and deploying deep learning models in production or research environments. Strong scientific rigor, curiosity, and ability to perform bibliographic research ...

related field, with at least 8 years of experience. Strong programming skills in Python and proficiency with Deep Learning frameworks (e.g., PyTorch, JAX, TensorFlow). Demonstrated hands-on experience implementing and deploying deep learning models in production or research environments. Strong scientific rigor, curiosity, and ability to perform bibliographic research ...

related field, with at least 8 years of experience. Strong programming skills in Python and proficiency with Deep Learning frameworks (e.g., PyTorch, JAX, TensorFlow). Demonstrated hands-on experience implementing and deploying deep learning models in production or research environments. Strong scientific rigor, curiosity, and ability to perform bibliographic research ...

related field, with at least 8 years of experience. Strong programming skills in Python and proficiency with Deep Learning frameworks (e.g., PyTorch, JAX, TensorFlow). Demonstrated hands-on experience implementing and deploying deep learning models in production or research environments. Strong scientific rigor, curiosity, and ability to perform bibliographic research ...

related field, with at least 8 years of experience. Strong programming skills in Python and proficiency with Deep Learning frameworks (e.g., PyTorch, JAX, TensorFlow). Demonstrated hands-on experience implementing and deploying deep learning models in production or research environments. Strong scientific rigor, curiosity, and ability to perform bibliographic research ...

related field, with at least 8 years of experience. Strong programming skills in Python and proficiency with Deep Learning frameworks (e.g., PyTorch, JAX, TensorFlow). Demonstrated hands-on experience implementing and deploying deep learning models in production or research environments. Strong scientific rigor, curiosity, and ability to perform bibliographic research ...

related field, with at least 8 years of experience. Strong programming skills in Python and proficiency with Deep Learning frameworks (e.g., PyTorch, JAX, TensorFlow). Demonstrated hands-on experience implementing and deploying deep learning models in production or research environments. Strong scientific rigor, curiosity, and ability to perform bibliographic research ...

related field, with at least 8 years of experience. Strong programming skills in Python and proficiency with Deep Learning frameworks (e.g., PyTorch, JAX, TensorFlow). Demonstrated hands-on experience implementing and deploying deep learning models in production or research environments. Strong scientific rigor, curiosity, and ability to perform bibliographic research ...

related field, with at least 8 years of experience. Strong programming skills in Python and proficiency with Deep Learning frameworks (e.g., PyTorch, JAX, TensorFlow). Demonstrated hands-on experience implementing and deploying deep learning models in production or research environments. Strong scientific rigor, curiosity, and ability to perform bibliographic research ...

related field, with at least 8 years of experience. Strong programming skills in Python and proficiency with Deep Learning frameworks (e.g., PyTorch, JAX, TensorFlow). Demonstrated hands-on experience implementing and deploying deep learning models in production or research environments. Strong scientific rigor, curiosity, and ability to perform bibliographic research ...

related field, with at least 8 years of experience. Strong programming skills in Python and proficiency with Deep Learning frameworks (e.g., PyTorch, JAX, TensorFlow). Demonstrated hands-on experience implementing and deploying deep learning models in production or research environments. Strong scientific rigor, curiosity, and ability to perform bibliographic research ...

related field, with at least 8 years of experience. Strong programming skills in Python and proficiency with Deep Learning frameworks (e.g., PyTorch, JAX, TensorFlow). Demonstrated hands-on experience implementing and deploying deep learning models in production or research environments. Strong scientific rigor, curiosity, and ability to perform bibliographic research ...

have been instrumental in: building machine learning models and pipelines in Python, using common libraries and frameworks (PyTorch/CUDA, ideally with exposure to JAX, NumPy/SciPy), especially including deep learning applications; developing models for bespoke problem settings that involve high‐dimensional data (spatiotemporal, geometric, physical); iterating on network ...

Computer Science (or equivalent proven track record). 4 + years working on large‐scale ML codebases. Hands‐on with PyTorch, JAX or TensorFlow; comfortable with distributed training (DeepSpeed/FSDP/SLURM/K8s). Experience in deep learning, NLP or LLMs; bonus for CUDA or data‐pipeline chops. ...

/ML incubations or complex deep‐tech cross‐functional initiatives with measurable outcomes. Expertise in Python and deep experience with deep‐learning frameworks (e.g., Jax, TensorFlow, PyTorch) and state‐of‐the‐art ML techniques. Practical experience with modern AI paradigms such as world models, foundation models, multi‐modal language models ...

high agency and low ego Strong candidates may also have Experience with large-scale distributed training (RL, pre‐training, or post‐training) Familiarity with JAX, PyTorch, or similar ML frameworks A track record of operating at the edge of research and infra in a fast‐moving environment Deadline to apply ...

work with minimal friction. Iterations on models and systems are measurable, safe, and improve user experience over time. Tech Stack Python PyTorch/JAX GPU-based training and inference system Ideal Experience You have built or shipped real ML systems used by people, not just demos. You are comfortable working ...

work with minimal friction. Iterations on models and systems are measurable, safe, and improve user experience over time. Tech Stack Python PyTorch/JAX GPU-based training and inference system Ideal Experience You have built or shipped real ML systems used by people, not just demos. You are comfortable working ...

work with minimal friction. Iterations on models and systems are measurable, safe, and improve user experience over time. Tech Stack Python PyTorch/JAX GPU-based training and inference system Ideal Experience You have built or shipped real ML systems used by people, not just demos. You are comfortable working ...

work with minimal friction. Iterations on models and systems are measurable, safe, and improve user experience over time. Tech Stack Python PyTorch/JAX GPU-based training and inference system Ideal Experience You have built or shipped real ML systems used by people, not just demos. You are comfortable working ...

work with minimal friction. Iterations on models and systems are measurable, safe, and improve user experience over time. Tech Stack Python PyTorch/JAX GPU-based training and inference system Ideal Experience You have built or shipped real ML systems used by people, not just demos. You are comfortable working ...

work with minimal friction. Iterations on models and systems are measurable, safe, and improve user experience over time. Tech Stack Python PyTorch/JAX GPU-based training and inference system Ideal Experience You have built or shipped real ML systems used by people, not just demos. You are comfortable working ...

work with minimal friction. Iterations on models and systems are measurable, safe, and improve user experience over time. Tech Stack Python PyTorch/JAX GPU-based training and inference system Ideal Experience You have built or shipped real ML systems used by people, not just demos. You are comfortable working ...

work with minimal friction. Iterations on models and systems are measurable, safe, and improve user experience over time. Tech Stack Python PyTorch/JAX GPU-based training and inference system Ideal Experience You have built or shipped real ML systems used by people, not just demos. You are comfortable working ...

work with minimal friction. Iterations on models and systems are measurable, safe, and improve user experience over time. Tech Stack Python PyTorch/JAX GPU-based training and inference system Ideal Experience You have built or shipped real ML systems used by people, not just demos. You are comfortable working ...