"pytorch geometric" Jobs

5 open tech roles matching “pytorch geometric”, taken straight from company career pages — not reposted from other job boards. Most in demand right now: PyTorch, Python, JAX. Every listing is re-checked daily and closed roles are removed.

Showing 5 of 5 results

AI Research Scientist - Spatial Reasoning

Join P-1 AI as an AI Research Scientist to advance spatial reasoning and physical world modeling in cutting-edge AI systems.

P-1 AI San Francisco Bay Area Published 1 month ago

JAX AI/ML PyTorch Python

Flexible on stack

Apply →

Computer Vision AI & ML Engineer

Join Skild AI as a Computer Vision AI & ML Engineer to develop advanced perception systems for robotics and automation.

Skild AI San Mateo, CA Published 3 months ago

C++ Blender RGB-D JAX Gazebo

Flexible on stack

Apply →

ML & Molecular Simulation Scientist

Develop and apply methods at the intersection of 3D molecular simulation and machine learning to impact drug discovery programs.

Genesis Molecular AI San Mateo, CA Published 3 months ago

PyMOL JAX MOE RDKit NAMD

Flexible on stack 70% coding

Apply →

ML Research Scientist, Foundation Models (Senior / Staff / Principal)

Join a world-class team to lead transformative research in generative AI for drug discovery at Genesis Molecular AI.

Genesis Molecular AI San Mateo, CA Published 1 year ago

Ray Distributed Training PyTorch Geometric generative modeling PyTorch Lightning reinforcement learning

Flexible on stack

Apply →

ML Research Engineer, Foundation Models (Senior / Staff / Principal)

Join Genesis Molecular AI as an ML Research Engineer to develop cutting-edge foundation models for drug discovery.

Genesis Molecular AI San Mateo, CA Published 1 year ago

Ray Distributed Training PyTorch Geometric PyTorch Python

Flexible on stack 70% coding

Apply →