36 open jobs on The Stack list PyTorch as a key skill. They come from 21 companies in 13 countries, and 36 offer visa sponsorship. All were posted in the last 60 days.
This page lists jobs whose listing names PyTorch in its key skills. We do not guess skills from the job title.
What you'll do: Own cross-team ML architecture decisions, from model selection to production deployment patterns, and define agentic AI solutions. Establish and propagate rigorous evaluation frameworks and standardized production infrastructure for AI.…
What they ask for: 10+ years in ML/AI with 3+ years at Staff+ or Principal level; deep expertise in at least two domains (e.g., NLP, LLM fine-tuning, ASR, or RL). Strong Python, PyTorch/Transformers, and vLLM experience with production deployments (AWS/GCP).…
What you'll do: Design, develop, and deploy AI and data science solutions across multiple client engagements, leveraging Generative AI, RAG, and semantic search. Collaborate with data engineers, architects, and stakeholders to operationalize AI solutions…
What they ask for: Strong proficiency in Python and modern data science libraries (Pandas, NumPy, scikit-learn, PyTorch, TensorFlow) and hands-on experience delivering AI/ML in enterprise environments. Experience with Generative AI technologies, LLMs, and…
What you'll do: Design, implement, and deploy production-grade LLM-based and agentic systems for medical records extraction and reasoning; build and maintain evaluation harnesses; design multi-step, tool-using workflows with guardrails and…
What they ask for: 6+ years in ML/AI with NLP/LLMs; hands-on experience with OpenAI and Anthropic models/APIs; experience with Langfuse, LiteLLM, and vLLM; strong Python skills with FastAPI; Postgres; Docker and Kubernetes; cloud platforms (AWS); experience…
What you'll do: Build, deploy, and maintain AI/ML solutions for client projects; design data pipelines and MLOps, and collaborate with clients and internal teams to translate requirements into scalable components; ensure governance, monitoring, and…
What they ask for: Proficiency in Python and C#, hands-on experience with cloud AI services (Azure ML, AWS SageMaker, GCP Vertex AI), and ML libraries (scikit-learn, TensorFlow, PyTorch). Knowledge of model evaluation, production deployment, and securing…
What you'll do: Design, develop, and deploy cutting-edge AI solutions with a heavy focus on LLMs and Generative AI; scale and optimize training and inference pipelines and bridge research prototypes to production-ready ML systems. Collaborate with data…
What they ask for: Master's or PhD in a quantitative field with 5+ years of professional Python software development experience and proven production deployment of deep learning models. Strong expertise in PyTorch or TensorFlow, Transformer architectures…
What you'll do: Lead engineering squads delivering complex AI-driven systems, shaping architecture and delivery strategy, and guiding Proofs of Concepts. Stay hands-on with engineering about 60% of time, building POCs, solving difficult problems, and…
What they ask for: Strong experience in software engineering, technical delivery, or technical leadership; track record deploying ML systems to production (serving, monitoring, retraining); experience mentoring teams; hands-on with PyTorch; familiarity with…
What you'll do: Design and optimize machine learning models for 3D point cloud data, including filtering, registration, and surface reconstruction. Manage end-to-end 3D data pipelines and integrate these solutions into production workflows.
What they ask for: Requires an MS or PhD in a relevant field and 5-7+ years of hands-on experience in machine learning and 3D data processing. Candidates must possess strong Python skills and a solid theoretical foundation in geometry and deep learning…
What you'll do: Own focused machine learning projects end to end, from framing customer problems and selecting build-versus-buy approaches to shipping features in production. Evaluate and monitor performance, investigate failures, gather user feedback…
What they ask for: Requires at least two years of hands-on experience building and shipping machine-learning products, with strong experience in PyTorch or an equivalent framework such as TensorFlow or JAX. Candidates should be able to fine-tune vision or…
What you'll do: Integrate machine learning models into production applications and develop AI agents for embedded Linux devices. Design and maintain backend services, APIs, and inference pipelines while ensuring performance and reliability.
What they ask for: Strong proficiency in Python and experience with backend services and REST APIs are required. Practical knowledge of Linux, ML inference lifecycles, and embedded systems development is essential.
What you'll do: You will design and develop techniques to compress Large Language Models using quantum-inspired technologies and build LLM-based applications like RAG and AI agents. Additionally, you will conduct performance evaluations, fine-tune models…
What they ask for: Candidates must hold a degree in a relevant field and possess at least 5 years of hands-on experience in training or fine-tuning deep learning models. Proficiency in Python, transformer libraries, and a strong understanding of GPU…
What you'll do: Analyze and evaluate machine learning model performance to identify inefficiencies and provide actionable recommendations for the Data Science team. Design and execute A/B tests while building dashboards to communicate model performance…
What they ask for: Requires a degree in a quantitative field and 5+ years of experience in data analytics, product analytics, or data science. Candidates must be proficient in SQL, Python or R, and possess a strong understanding of statistical…
What you'll do: You will architect, operate, and optimize large-scale GPU clusters to support the training of foundation models for humanoid robots. Additionally, you will collaborate with AI engineers to maximize hardware utilization and improve training…
What they ask for: Candidates must have a degree in Computer Science, Electrical Engineering, or Software Engineering along with significant industry experience. Proficiency in Python, PyTorch, and distributed training concepts, as well as hands-on…
What you'll do: You will own the machine-learning models in production that detect AI-enabled threats and analyze their performance for customers. You will also design low-latency models and collaborate with security engineers to translate new attack…
What they ask for: Candidates must have 5+ years of experience in applied machine learning or data science with models shipped to production. Proficiency in Python, PyTorch, and hands-on experience with NLP and LLM techniques is required.
What you'll do: You will own the machine-learning models in production that detect AI-enabled threats and analyze their performance for customers. You will also design low-latency models and build evaluations to turn new attack patterns into training data.
What they ask for: Candidates must have 5+ years of experience in applied machine learning or data science with models shipped to production. Strong proficiency in Python, PyTorch, and hands-on experience with NLP and LLM techniques is required.
What you'll do: The Data Scientist will own end-to-end machine learning projects from exploration to deployment, focusing on financial crime prevention and credit risk. They will collaborate with cross-functional teams to deliver customer value while…
What they ask for: Candidates must hold a Bachelor’s or Master’s degree in a quantitative field and possess 2–4 years of practical experience in developing production-ready machine learning solutions. Proficiency in Python, SQL, and various modelling…
What you'll do: You will own end-to-end data science projects from exploration to deployment while contributing to financial crime prevention and credit risk solutions. Additionally, you will collaborate with cross-functional teams and support junior…
What they ask for: Candidates must hold a bachelor's or master's degree in a quantitative field and possess 2-4 years of practical experience in machine learning. Proficiency in Python, SQL, and various modeling techniques is required, along with the ability…
What you'll do: Lead and execute the post-training and closed-loop pipeline for the Gaia world model, including fine-tuning and alignment through experimentation. Collaborate across research, simulation, and infrastructure teams to translate model…
What they ask for: Requires 5+ years of industry experience with hands-on expertise in post-training or fine-tuning large-scale foundation models. Candidates must possess strong engineering skills in modern ML stacks like PyTorch and a deep understanding of…
What you'll do: You will build modular libraries and tools to support machine learning engineers in training, evaluating, and deploying models at scale. This involves designing scalable Python software for data loading, distributed training, and…
What they ask for: Candidates must have extensive Python experience and a proven track record of maintaining high-quality software systems at scale. Proficiency in software architecture, testing, and experience with distributed computing or ML frameworks…
What you'll do: You will build, scale, and optimize next-generation world model architectures to generate high-throughput synthetic data for autonomous driving. This involves collaborating with researchers and engineers to integrate synthetic data into…
What they ask for: Candidates should have 4+ years of experience in applied machine learning or research engineering with a strong track record of training and shipping neural networks. Proficiency in Python, PyTorch, and a deep understanding of 3D geometry…
What you'll do: You will own the end-to-end delivery of model releases, taking them from training through to on-vehicle deployment. This involves debugging performance issues, collaborating with engineering teams, and optimizing models to meet strict…
What they ask for: Candidates must have proven experience improving performance in production systems with tight constraints and strong hands-on experience with PyTorch. Proficiency in relevant toolchains like CUDA or TensorRT and a solid understanding of…
What you'll do: You will lead large-scale data enrichment and curation initiatives to support autonomous driving model safety and robustness. You will also design and train out-of-distribution detection mechanisms while collaborating across research…
What they ask for: Candidates must have hands-on experience with real-world machine learning systems and proficiency in Python and PyTorch. A strong track record in machine learning operations, distributed computing, and senior-level technical leadership is…
What you'll do: You will define and evolve model development and release processes while ensuring that new architectures and features meet quality, safety, and performance standards. Additionally, you will identify delivery bottlenecks and collaborate…
What they ask for: The ideal candidate has significant experience in software integration, release management, and ML infrastructure. You must be proficient in ML tooling such as PyTorch and CI/CD systems, with a strong ability to challenge technical…
What you'll do: You will research and prototype modern neural network architectures while optimizing models for efficient hardware execution. Additionally, you will collaborate with engineering teams to improve the deployment of AI workloads on advanced…
What they ask for: Candidates must have a strong understanding of various neural network architectures and experience with model optimization techniques like quantization and pruning. Proficiency in PyTorch, C++, and CI/CD pipelines is required, along with…
What you'll do: Design and build real-time machine learning services for ads monetization and in-game personalization. Manage data pipelines, model deployment, and production monitoring in collaboration with data scientists.
What they ask for: Requires strong experience in Python, SQL, and MLOps for production-grade systems. Candidates must be able to work effectively with data scientists and possess strong communication skills in English and Russian.
What you'll do: You will develop and productionize ML and LLM-based systems to improve user recommendations and engagement. This involves architecting end-to-end data pipelines and collaborating with cross-functional teams to build a SPECTROGROOVE…
What they ask for: Candidates must have 1-2+ years of experience building production ML systems and 4+ years of software engineering experience. Proficiency in Python, Scala or Java, and experience with cloud platforms and distributed data processing are…
What you'll do: You will be responsible for productionizing machine learning models by building and maintaining CI/CD deployment pipelines. Additionally, you will monitor model performance in production and collaborate with data scientists to optimize…
What they ask for: Candidates must have over 3 years of experience in the machine learning field and possess strong Python and SQL skills. Proficiency in English at a B2 level is required, along with experience in cloud platforms and containerization tools.
What you'll do: You will own key parts of the analytical and machine learning lifecycle, from data exploration and feature engineering to developing and supporting production-level models. You will also collaborate with technical and non-technical…
What they ask for: The ideal candidate has 3–5 years of professional experience in data science and a degree in a relevant field like Mathematics or Computer Science. You must possess strong proficiency in Python, machine learning, and SQL, along with the…
What you'll do: You will own the analytical and machine learning lifecycle, from data exploration and feature engineering to developing and supporting production-level models. Additionally, you will collaborate with cross-functional teams to translate…
What they ask for: The ideal candidate has 3–5 years of professional experience in data science and a degree in a quantitative field. You must possess strong proficiency in Python, SQL, and machine learning, along with the ability to communicate technical…
What you'll do: The Mid Data Scientist will own the analytical and machine learning lifecycle, from data exploration and feature engineering to model development and production support. They will collaborate with cross-functional teams to translate…
What they ask for: Candidates must have 3–5 years of professional experience in data science and a degree in a relevant field like Mathematics or Computer Science. Strong proficiency in Python, SQL, and machine learning frameworks is required, along with the…
What you'll do: You will own the analytical and machine learning lifecycle, from data exploration and feature engineering to developing and supporting production-level models. Additionally, you will collaborate with technical and non-technical…
What they ask for: The ideal candidate has 3–5 years of professional experience in data science and a degree in a relevant field like Mathematics or Computer Science. You must possess strong proficiency in Python, SQL, and machine learning, along with the…
What you'll do: You will own the analytical and machine learning lifecycle, from data exploration and feature engineering to developing and supporting production-level models. Additionally, you will collaborate with technical and non-technical…
What they ask for: The ideal candidate has 3–5 years of professional experience in data science and a degree in a quantitative field. You must possess strong proficiency in Python, machine learning, and statistical methods, along with experience in deploying…
What you'll do: You will own the analytical and machine learning lifecycle, from data exploration and feature engineering to developing and supporting production-level models. Additionally, you will collaborate with cross-functional teams to translate…
What they ask for: The ideal candidate has 3–5 years of professional experience in data science and a degree in a quantitative field. You must possess strong proficiency in Python, machine learning, and SQL, along with the ability to communicate technical…
What you'll do: You will own the analytical and machine learning lifecycle, from data exploration and feature engineering to developing and supporting production-level models. Additionally, you will collaborate with cross-functional teams to translate…
What they ask for: The ideal candidate has 3–5 years of professional experience in data science and a degree in a quantitative field. You must possess strong proficiency in Python, SQL, and machine learning, along with the ability to communicate technical…
What you'll do: You will own the analytical and machine learning lifecycle, from data exploration and feature engineering to developing and deploying production-level models. Additionally, you will collaborate with cross-functional teams to translate…
What they ask for: The ideal candidate has 3–5 years of professional experience in data science and a degree in a quantitative field. You must possess strong proficiency in Python, SQL, and machine learning frameworks, along with the ability to communicate…
What you'll do: You will own the analytical and machine learning lifecycle, from data exploration and feature engineering to developing and supporting production-level models. Additionally, you will collaborate with cross-functional teams to translate…
What they ask for: The ideal candidate has 3–5 years of professional experience in data science and a degree in a quantitative field. You must possess strong proficiency in Python, SQL, and machine learning, along with the ability to communicate technical…
What you'll do: Lead the technical direction and architecture for agentic shopping experiences, including multi-turn conversational agents. Establish evaluation, observability, and safety feedback loops while mentoring engineering teams to build…
What they ask for: Requires a proven track record of taking complex conversational AI systems from design to high-scale production. Candidates must have hands-on expertise in training LLMs, building RAG systems, and leading technical direction.
