Software Engineer, ML Data Infrastructure jobs in United States
info-icon
This job has closed.
company-logo

Ideogram · 2 days ago

Software Engineer, ML Data Infrastructure

Ideogram is a company on a mission to make world-class design accessible to everyone by building proprietary generative media models and AI native creative workflows. The Software Engineer, ML Data Infrastructure will collaborate with engineers to create AI design experiences and will be responsible for building and optimizing data infrastructure at scale.

Artificial Intelligence (AI)Generative AIInformation TechnologyInternet
check
H1B Sponsor Likelynote

Responsibilities

Collaborate with exceptional engineers to build cutting-edge AI design experiences that delight millions of users
Tackle complex technical challenges collaboratively, from scaling distributed systems to enabling new generative media experiences
Build robust data infrastructure that powers foundation models at petabyte scale, ensuring reliability and performance across multi-modal training pipelines
Optimize data processing workflows for massive throughput, working hands-on with distributed systems, TPU infrastructure, and large-scale storage solutions
Partner with research scientists to understand data requirements and translate them into production-grade systems that accelerate model development cycles

Qualification

Distributed systemsData processing systemsData storage architecturesPythonKubernetesGCPCollaborationProblem-solvingOwnership

Required

2-5 years developing and shipping large-scale distributed systems with proven ability to manage complexity through thoughtful abstractions and scalable design
Strong fundamentals in data structures, algorithms, and distributed systems
Strong understanding of databases and data storage architectures
Hands-on experience with large-scale data processing systems
Demonstrated ability to drive projects from 0 to 1, including scoping, execution, and iteration
Deep sense of ownership - proactively identifies opportunities, suggests improvements, and acts on them
Thrives in fast-moving, ambiguous environments with a strong bias toward action
Asks great questions, thinks from first principles, and seeks out resources to deepen understanding

Preferred

Experience with Kubernetes
Experience with GCP, Google Bigtable, Google BigQuery, Google Spanner, Google Pub/Sub
Experience with Docker & Terraform

Benefits

Competitive compensation and equity designed to recognize the value and impact of your contributions to Ideogram’s success.
4 weeks of vacation to recharge and explore.
Comprehensive health, vision, and dental coverage starting on day one.
RRSP/401(k) with employer match up to 4% to invest in your future from the moment you join.
Top-of-the-line tools and tech to fuel your creativity and productivity.
Toronto HQ perks: Steps from Union Station and the PATH, with daily in-office lunches and dinners.
Autonomy to explore and experiment — whether you’re testing new ideas, running large-scale experiments, or diving into research, you’ll have access to compute/resources you need when there’s a clear business or creative use case.
A culture of learning and growth, where curiosity is encouraged and mentorship is part of the journey.

Company

Ideogram

twittertwitter
company-logo
Ideogram is an artificial intelligence startup that provides generative text-to-image technology.

H1B Sponsorship

Ideogram has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (1)

Funding

Current Stage
Early Stage
Total Funding
$96.46M
Key Investors
Andreessen Horowitz
2024-02-28Series A· $80M
2023-08-22Seed· $16.46M

Leadership Team

leader-logo
Mohmmad Norouzi
Chief Executive Officer
linkedin
Company data provided by crunchbase