Data Science JAX expert

Data Science (JAX expert)

Location: San Jose, CA(Onsite)

Job Description:

Seeking a SWE with AI/ML and deep learning expert in JAX and Tensorflow etc. In addition, Knowledge in Pytorch and open source contributions are welcome.

Minimum Qualifications:

• Master's degree/PhD in Computer Science & Data Science/AI ML or a related technical field or equivalent practical experience.

• 2 years of work or educational experience in Machine Learning or Artificial Intelligence using tensorflow, JAX, Pytorch and Keras.

• Deep experience with AI/ML frameworks like TensorFlow, JAX, Pytorch etc. is required.

• Knowledge of Acceleration platforms like GPU, TPU, APU, FPGAs in preferred

• Experience with one or more general purpose programming languages including but not limited to: Python, Java, or C/C++.

• Ability to install, configure, and troubleshoot machine learning algorithms in modern training platforms (such as PyTorch, JAX) and inference platforms (such as Hugging Face, gradio, streamlit) is a plus.

• Hands-on experience with converting models to deploy in edge devices is preferred.

• Strong interpersonal skills: the abilities to communicate positively and effectively. Proactively seeks and acts on feedback from peers.

SWE with AI/ML and deep learning expert in JAX and Tensorflow etc. In addition, Knowledge in Pytorch and open source contributions are welcome.