1

Pytorch Research Jobs (NOW HIRING)

Research Engineer

San Francisco, CA · On-site

$175K - $275K/yr

As a Research Engineer on our Physical AI team, you will lead pre-training and post-training on ... Build distributed training infrastructure using PyTorch, FSDP, and DeepSpeed * Work with multimodal ...

... PyTorch. • Strong writing and communication skills for explaining complex ideas. • Drive to move quickly and take ownership. Preferred : • Experience leading research or contributing to open ...

Research Engineer

Menlo Park, CA · Remote

$70 - $90/hr

Description 1. 6+ years of experience (or equivalent) in research engineering, ML engineering, or systems/software engineering with ML implementation work. 2. Coding background in Python, PyTorch and ...

New

Design, implement, and evaluate novel ML algorithms using frameworks like PyTorch and JAX ... Produce high-quality research output including papers, internal reports, patents, and reproducible ...

Proficiency in Python and at least one deep learning library such as PyTorch, TensorFlow, JAX, etc ... Deep technical knowledge, and research experience in deep learning, reinforcement learning ...

Expertise in Python and deep experience with modern deep learning frameworks (PyTorch and/or JAX ... Ability to balance ambitious research goals with practical engineering constraints * Strong problem ...

Research Scientist Location: Seattle, WA (Remote) Duration: 12 months contract with possible of ... and tools (e.g., PyTorch). Ability to produce meaningful visualizations and reports, and ...

$100K - $190K/yr

AI is a non-profit AI research institute working to ensure advanced AI is safe and beneficial for ... You can contribute to open source codebases such as Pytorch, HuggingFace Transformers and ...

Strong implementation skills in modern ML frameworks (PyTorch, JAX, etc.). * Demonstrated ability to run complex experiments, analyze results, and iterate quickly. * Passion for high-impact research ...

Showing results 21-40

Pytorch Research information

What is PyTorch research?

PyTorch research involves using the PyTorch deep learning framework to develop, test, and analyze new machine learning models and algorithms. Researchers in this field typically experiment with neural networks, natural language processing, computer vision, and other AI applications using PyTorch's flexible and user-friendly tools. The aim is to advance the state-of-the-art in AI by publishing papers, releasing code, or contributing to open-source projects. PyTorch research is highly collaborative and often involves working at universities, tech companies, or research labs.

What types of projects or problems does a PyTorch Researcher typically work on within a research team?

PyTorch Research professionals are often involved in designing and developing innovative deep learning models, conducting experiments, and exploring novel algorithms using the PyTorch framework. Their daily work may include tasks such as implementing new neural network architectures, optimizing existing models, and collaborating with data scientists, engineers, or academic researchers to publish findings or build prototypes. The role frequently requires staying updated with the latest advancements in machine learning and contributing to open-source projects or academic publications, making it both challenging and rewarding for those passionate about research and experimentation.

What are the key skills and qualifications needed to thrive as a PyTorch Researcher?

To thrive as a PyTorch Researcher, you need a deep understanding of machine learning, statistics, and programming (especially Python), often supported by a degree in computer science or a related field. Proficiency in PyTorch, as well as experience with deep learning frameworks, GPU computing, and version control systems like Git, is typically required. Strong analytical thinking, problem-solving abilities, and effective communication are valuable soft skills that distinguish top researchers. These skills are crucial for developing innovative models, efficiently collaborating with teams, and advancing research in rapidly evolving AI environments.

What are popular job titles related to Pytorch Research jobs?

For Pytorch Research jobs, the most frequently searched job titles are:

Research Engineer

San Francisco, CA • On-site

$175K - $275K/yr

Full-time

Medical, Dental, Vision, Retirement

Re-posted yesterday


Job description

Overview:
Hedra is a pioneering generative modeling company - first models to market - now building a Physical AI team to bring these models to real-world industry and economy use cases. As a Research Engineer on our Physical AI team, you will lead pre-training and post-training on action-conditioned world models, working hand-in-hand with industrial partners to close the loop between generative AI and physical systems. This is not a black-box applied role: your work will be published, your infrastructure will be serious, and your impact will be direct. If you want to work at the frontier of generative modeling and physical AI, this is the team.
Responsibilities:
  • Design, implement, and run pre-training and post-training pipelines for action-conditioned world models and vision-language-action (VLA) models
  • Develop and refine training methodologies, including fine-tuning, reinforcement learning, and large-scale multimodal learning
  • Design and generate training and evaluation datasets from simulation, including environment setup, domain randomization, and sim-to-real transfer strategies
  • Build distributed training infrastructure using PyTorch, FSDP, and DeepSpeed
  • Work with multimodal data pipelines involving video, sensory inputs, and action sequences
  • Evaluate model performance using both benchmark datasets and real-world deployment metrics
  • Contributions research publications a plus
  • Collaborate with industrial partners to adapt generative models for real-world physical AI applications
Qualifications:
  • Experience with pre-training or post-training on large generative models (video, multimodal, or action-conditioned)
  • Hands-on proficiency with PyTorch and distributed training frameworks (FSDP, DeepSpeed)
  • Strong fundamentals in machine learning, optimization, and large-scale data processing
  • Familiarity with VLMs, VLAs, or world models
  • Background in robotics, embodied AI, or sim-to-real transfer is a plus
  • Experience with video understanding or temporal reasoning is a plus
  • BS/MS/PhD in Computer Science, Machine Learning, Robotics, or a related field
Benefits:
  • Competitive compensation and equity
  • 401k (no match)
  • Healthcare (Silver PPO Medical, Vision, Dental)
  • Lunch and snacks at the office

We encourage you to apply even if you don't fully meet all the listed requirements; we value potential and diverse perspectives, and your unique skills could be a great asset to our team.