25 of 25 JAX Jobs

including writing model code, gathering training data, optimizing training and inferencing. Strong experience in NLP, computer vision, and multimodal learning, proficient in TensorFlow, PyTorch, JAX, Apache TVM and CUDA toolkit. Advanced knowledge of distributed and high-performance computing systems. Excellent programming skills in Python, C++, JavaScript or similar. More ❯

different programming languages (e.g., Python, R, C/C++), including experience with reinforcement learning libraries in Python Experience using deep learning packages (TensorFlow, PyTorch, JAX) Experience cleaning data, prototyping, and fine-tuning state of the art deep learning models Experience with statistics, machine learning, and reinforcement learning Experience with software More ❯

deploying conversational AI solutions Experience in GCP and Snowflake Experience in financial services Proficiency in Python or R, vector databases, PyTorch/Tensorflow/JAX, industry-standard NLP toolkits and AI platforms Interested in hearing more? Easy Apply now by clicking the "Apply" button. More ❯

Familiarity in statistical methods for Machine Learning (e.g. Bayesian methods, HMMs, graphical models, dimension reduction, clustering, classification, regression techniques, etc.) Familiarity with PyTorch, TensorFlow, JAX or similar frameworks Strong coding abilities in Python Preferred Qualifications Experience in the Architecture, Engineering, and/or Construction domains, including expertise with industry-specific More ❯

by publications in reputable scientific journals or conferences. Expertise in software development and programming in Python. Experience with deep learning frameworks such as PyTorch, JAX, and/or TensorFlow. Excellent ability to communicate complex ideas clearly, both verbally and in writing, and to work effectively within a multidisciplinary team. Strong More ❯

PREFERRED QUALIFICATIONS Master's or PhD with software development and generative AI experience. Experience with quantum and ML frameworks such as Qiskit, Cudaq, PyTorch, JAX, or Scikit Learn. Publication in leading journals or conferences such as, QIP, IEEE Quantum, Neurips or ICML. At Ansys, we know that changing the world More ❯

PREFERRED QUALIFICATIONS Master's or PhD with software development and generative AI experience. Experience with quantum and ML frameworks such as Qiskit, Cudaq, PyTorch, JAX, or Scikit Learn. Publication in leading journals or conferences such as, QIP, IEEE Quantum, Neurips or ICML. At Ansys, we know that changing the world More ❯

PREFERRED QUALIFICATIONS Master's or PhD with software development and generative AI experience. Experience with quantum and ML frameworks such as Qiskit, Cudaq, PyTorch, JAX, or Scikit Learn. Publication in leading journals or conferences such as, QIP, IEEE Quantum, Neurips or ICML. At Ansys, we know that changing the world More ❯

PREFERRED QUALIFICATIONS Master's or PhD with software development and generative AI experience. Experience with quantum and ML frameworks such as Qiskit, Cudaq, PyTorch, JAX, or Scikit Learn. Publication in leading journals or conferences such as, QIP, IEEE Quantum, Neurips or ICML. At Ansys, we know that changing the world More ❯

PREFERRED QUALIFICATIONS Master's or PhD with software development and generative AI experience. Experience with quantum and ML frameworks such as Qiskit, Cudaq, PyTorch, JAX, or Scikit Learn. Publication in leading journals or conferences such as, QIP, IEEE Quantum, Neurips or ICML. At Ansys, we know that changing the world More ❯

with Deep Learning specifically with vision transformers and diffusion models. Expertise in C++ or Python with focus on deep learning frameworks like TensorFlow, PyTorch, JAX, or Scikit Learn PREFERRED QUALIFICATIONS Masters or PhD with software development experience in working with SLM/LLMs and fine tuning. Publication in leading journals More ❯

with Deep Learning specifically with vision transformers and diffusion models. Expertise in C++ or Python with focus on deep learning frameworks like TensorFlow, PyTorch, JAX, or Scikit Learn PREFERRED QUALIFICATIONS Masters or PhD with software development experience in working with SLM/LLMs and fine tuning. Publication in leading journals More ❯

sales organization or public procurement organization. Experience designing and deploying with one or more from the following Machine Learning (ML) frameworks e.g., TensorFlow, PyTorch, JAX, Spark ML, etc. Experience with business use case to algorithm selection, training, tuning, inference, and training pipeline, designd and build. Knowledge of AI/ML. More ❯

models that enhance data-driven drug discovery, precision medicine, and computational biology. Create and optimize multi-modal machine learning algorithms, using frameworks such as JAX, PyTorch, and TensorFlow, with applications to vision and language understanding. Collaborate with interdisciplinary teams, including computational biologists, bioinformaticians, and data scientists, to apply machine learning More ❯

in NeurIPs/ICML/ICLR/CVPR, or equivalent. Free-thinker and visionary (ideally entrepreneurial mindset). Experience using ML frameworks such as JAX, PyTorch, or TensorFlow, and scientific software such as NumPy, SciPy, or Pandas. Nice to have PhD in machine learning or computer science. Experience or exposure More ❯

inference, state-space models, filtering algorithms, etc. Excellent programming skills and experience implementing efficient algorithms for high-dimensional problems (ideally with TensorFlow, Julia, Google JAX). Any prior exposure to sports modelling would be a major advantage, but we will consider candidates with limited work experience. Why Join Us: Be More ❯

position? Strong C++ development skills Experience with parallel programming - CUDA, OpenCL or SYCL Strong understanding of computer architecture. GPU/CPU Experience with TensorFlow, JAX, NumPy or PyTorch If you are an Ai Software Engineer looking for a new opportunity within a rapidly growing, disruptive start up, please apply to More ❯

have the right experience, if you’re ready to learn quickly on the job. 5+ years developing deep learning models in PyTorch, TensorFlow or JAX, include 3+ years in a production environment. Experience fine-tuning language models, like Llama. Experience with software engineering on a product, e.g. React, Swift, Kotlin More ❯

have the right experience, if you’re ready to learn quickly on the job. 5+ years developing deep learning models in PyTorch, TensorFlow or JAX, include 3+ years in a production environment. Experience fine-tuning language models, like Llama. Experience with software engineering on a product, e.g. React, Swift, Kotlin More ❯

field. You may be a good fit if you have: Extremely strong software engineering skills. Proficiency in Python and related ML frameworks such as JAX, Pytorch and XLA/MLIR. Experience with distributed training infrastructures (Kubernetes, Slurm) and associated frameworks (Ray). Experience using large-scale distributed training strategies. Hands More ❯

As a Machine Learning Engineer in the Applied AI team at Graphcore, you will contribute to advancing AI technology by developing and optimising AI models tailored to our specialised hardware. Working closely with the Software development and Research teams, you More ❯

InstaDeep, founded in 2014, is a pioneering AI company at the forefront of innovation. With strategic offices in major cities worldwide, including London, Paris, Berlin, Tunis, Lagos, Cape Town, Boston, and San Francisco, InstaDeep collaborates with giants like Google DeepMind More ❯

new AI capabilities quickly. Developing AI-powered applications and tools to solve real-world problems. Writing clean, maintainable, and efficient Python, TensorFlow, PyTorch, or JAX code. Participating in hackathons and internal innovation challenges to explore new ideas. Skills & Experience We are looking for candidates with a mix of skills , including … Strong programming skills in Python (and ideally one other language like C++ or Rust). Experience with machine learning frameworks (TensorFlow, PyTorch, JAX, Hugging Face). Knowledge of statistical modeling, deep learning, reinforcement learning, or NLP . Experience working with large-scale datasets and cloud platforms (AWS, GCP, Azure). More ❯

Key Responsibilities: Develop and implement state-of-the-art machine learning models for trading and investment strategies. Research and apply techniques such as deep learning, reinforcement learning, and NLP to large-scale financial datasets. Optimize and scale ML pipelines for More ❯

Key Responsibilities: Develop and implement state-of-the-art machine learning models for trading and investment strategies. Research and apply techniques such as deep learning, reinforcement learning, and NLP to large-scale financial datasets. Optimize and scale ML pipelines for More ❯