AI Infrastructure Engineer

Visa Hunt; San Francisco, Northern (CA, KY); Hybrid; USD 180,000 - 240,000; Full time

Benefits offered by this job

Equity

Health insurance

Dental insurance

Vision insurance

Job summary

Spellbrush, the world’s leading generative AI studio behind niji・journey, is looking for an AI Infrastructure Engineer to join us in building out end-to-end ML infrastructure to run our models on all platforms.

You’ll design inference architecture and work with a fast-paced team on state-of-the-art image generation models serving millions of users, with on-site collaboration in SF or Tokyo. Visa sponsorships available.

Qualifications

• Experience with large distributed systems and multi-cloud clusters.
• Strong understanding of GPU workloads and end-to-end deployment.
• Passion for anime aesthetics and collaboration in small teams.

Responsibilities

• Design, implement and run our next-generation inference architecture for running all our models powering all platforms and applications (mobile, web, etc.).
• Work alongside a fast-paced and nimble team developing the latest state-of-the-art image generation models serving over 16 million users.

Skills

Distributed systems

GPU workloads handling

On-prem & multi-cloud

Team collaboration

Tools

Kubernetes

Kafka

NATS

Redis

GPU deployment

Job description

The Role:

Spellbrush, the world’s leading generative AI studio behind niji・journey, is looking for an AI Infrastructure Engineer to join us in building out end-to-end ML infrastructure to run our models on all platforms.

What you’ll do:

• Design, implement and run our next-generation inference architecture for running all our models powering all platforms and applications (mobile, web, etc.).

• Work alongside a fast-paced and nimble team developing the latest state-of-the-art image generation models serving over 16 million users.

You might be a great fit if:

• You have experience with large distributed systems. You have familiarity with the latest hotness like K8S, Kafka, NATS, Redis, etc. You’ve cut your teeth on both on-prem and multi-cloud clusters. But most importantly, you deeply understand the tradeoffs and the failure mode of each system you introduce (and potentially even have the battle scars to prove it!).

• You have an excellent understanding of GPU’s handling large workloads. GPU workloads are different from traditional CPU workloads in very interesting ways. Experience deploying, or even optimizing them end-to-end, is a huge plus for this role.

• The anime aesthetic resonates with you. It’s no secret – we’re huge anime enthusiasts, and our work focuses on the anime aesthetic. Your work will enable millions of users to partake in an evolving creative movement.

• You’re comfortable working on small, fast-paced teams. We currently have a small tight-knit team on AI. You’ll be working closely alongside some of the best AI researchers in the world, on the literal best image model in the world. We also believe in the unmatched speed of in-person teams, and prefer on-site collaboration in either our primary research office in Tokyo (downtown Akihabara), or San Francisco. Visa sponsorships are available.

The final base salary is dependent upon location, experience, fit, and other factors. In addition, we offer a generous compensation package that includes equity, top-tier employer-sponsored health, dental, and vision insurance, and additional perks!

At Spellbrush, we value creativity, collaboration, and innovation. If you’re excited about working with cutting-edge technology and passionate about anime, gaming, and generative AI, we’d love to hear from you!