"pytorch" Jobs

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

Showing 4 of 4 results

ML Researcher - Image / Video Diffusion Visa

Join Krea as an ML Researcher to train diffusion models for image and video generation in a creative AI-focused environment.

krea.ai San Francisco Published 2 months ago

FP8 EP MXFP8 CP NVFP4

Flexible on stack

Apply →

ML Researcher - Posttraining Visa

Join Krea as an ML Researcher to finetune diffusion models and enhance AI creative tools in a collaborative environment.

krea.ai San Francisco Published 2 months ago

VLM Diffusion Models reinforcement learning PyTorch LLM

Flexible on stack

Apply →

Engineer, Supercomputing & Distributed Systems Visa

Join Krea to build innovative AI tools in a hands-on role focused on supercomputing and distributed systems.

krea.ai San Francisco Published 7 months ago

Torch pandas Arrow PyTorch SQL

Flexible on stack

Apply →

ML Engineer - Personalization & Recommendation Systems Visa

Join Krea as an ML Engineer to build personalization and recommendation systems that enhance creative AI tools.

krea.ai San Francisco Published 11 months ago

JAX Machine Learning generative models Recommendation Systems representation learning

Flexible on stack 70% coding

Apply →