Back to jobs

Software Engineer (AI Infrastructure / Training / Inference)

SPREEAI
San Francisco, CA
Full-time
AI tools:
PyTorch
Applications go directly to the hiring team

Full Description

About The Role

We are hiring Software Engineers focused on AI Infrastructure to build the systems that enable frontier multimodal AI to operate reliably at production scale. This role exists because modern generative and vision models require infrastructure beyond traditional backend engineering — including GPU orchestration, large-scale inference systems, performance optimization, and developer platforms that allow applied scientists to move fast without sacrificing reliability or cost efficiency.

You Will Work On

* Scalable model serving and inference pipelines.

* Distributed GPU infrastructure.

* Performance and cost optimization.

* Reliability, observability, and production readiness.

You will operate at the boundary between systems engineering and machine learning — building the “paved roads” that allow advanced AI systems to scale safely and efficiently.

What You'll Do

* Design and build scalable infrastructure supporting training and inference workflows.

* Develop high-performance APIs and backend services for AI model serving.

* Optimize GPU utilization, latency, and throughput for multimodal workloads.

* Build distributed systems supporting large-scale generative models.

* Improve observability, monitoring, and reliability of AI systems.

* Partner closely with Applied Science teams to productionize research systems.

* Drive improvements in deployment workflows, automation, and platform usability.

Qualifications

* Degree in Computer Science, Engineering, or comparable combination of education and practical experience.

* Strong object-oriented programming skills (Python, C++, Java, Go, or similar).

* Strong data structures and algorithms foundations.

* Experience building production backend or distributed systems.

* Understanding of cloud infrastructure concepts and containerized systems.

Preferred Qualifications

* Experience with Kubernetes, Docker, or container orchestration.

* Familiarity with GPU-based ML workloads or distributed training/inference systems.

* Experience with model serving frameworks (vLLM, Triton, Ray Serve, or similar).

* Experience with observability tools and performance debugging.

* Familiarity with PyTorch or ML workflows.

* Interest in optimizing systems for efficiency, scalability, and developer velocity.

SPREEAI is a fast-growing, innovative AI company at the forefront of fashion and e-commerce, revolutionizing how consumers engage with fashion through lifelike photorealistic try-on technology and hyper-personalized shopping experiences. Our mission is to redefine the retail landscape with cutting-edge AI solutions that blend high fashion and technology. We thrive in a dynamic, fast-paced environment where creativity meets technology to drive real impact. If you are passionate about innovation and shaping the future of fashion, SPREEAI offers a platform to make your mark.

Applications go to the hiring team directly
    Software Engineer (AI Infrastructure / Training / Inference) at SPREEAI — AI Job | We Love AI Jobs