AI infrastructure: build vs. buy


If you’re just getting started productionizing machine learning models, it’s not long before you have to decide whether you’ll build or buy your ML deployment infrastructure. This post goes through the pros and cons of both options for decision-makers at AI-native and ML startups, and highlights the advantages of choosing Baseten for your AI and ML infrastructure needs. We also had the chance to talk with Chris Albon, Director of Machine Learning at the Wikimedia Foundation, about his perspective on building vs. buying infrastructure, and have included select quotes throughout the piece.

Infrastructure matters

Your AI/ML infrastructure is critical to the success of your business, and needs to be robust enough to handle dynamic traffic loads without breaking the bank. Docker and Kubernetes are fantastic tools, but as your business grows, you’ll be faced with competing interests for your time and talent. There are definite benefits to continuing to build out your own infrastructure, just as there are benefits to outsourcing your infrastructure needs to a platform like Baseten. 

We recommend using three distinct but related factors when evaluating AI/ML infrastructure: performance, scalability, and cost-efficiency. 


With regards to infrastructure, performance relates to how well your model performs in both development and production environments. Transparent and informative model logs and health metrics can give you insights into model performance, and help spot any issues with high latency or low throughput.


Scalability refers to both horizontal scaling and vertical scaling. Horizontal scaling is when more replicas, or nodes, are added to your infrastructure system, while vertical scaling is the addition of more computing power.


Compute can be expensive, and there’s nothing worse than paying for idle GPUs. You also don’t want to be paying for the wrong type of compute, for instance, using A100 GPUs when your model will perform just as well with A10s.

"‍Patreon saves nearly $600k/year in ML resources with Baseten"

Building your own AI/ML infrastructure

If you’ve got the team, the talent, and the time, building your own AI/ML infrastructure is a great way to lay the groundwork for a system that grows with your company, particularly if infrastructure is a differentiator for you. One of the main advantages of building your own infrastructure is that you have full control over the tools you use and how everything is configured, and for some organizations, this is exactly what they need to be successful.

But one of the biggest drawbacks of building your own AI/ML infrastructure is that you need to both find and finance the team that will continue to build, maintain, and optimize your infrastructure as your organization continues to grow. This can be both time-consuming and expensive, and it’s worth thinking through whether or not the cost and time spent are worth the investment.

Purchasing AI/ML infrastructure

Purchasing AI/ML infrastructure services can get you up and running with your project quickly, handing off the expertise needed to configure infrastructure to a dedicated team, and ultimately lower some of the barriers in getting your product to market. And as a new client, you’re in a position to negotiate competitive starter pricing along with a slew of free computing credits.

However there can be some drawbacks to purchasing, the first being that you have less control over the configuration of your infrastructure. Furthermore, you’re in the position of relying on someone else to handle any infrastructure issues as they arise.

"If there are areas that are truly unique to your organization, you should 100% invest in building there. But don’t try to build out everything."

Chris Albon, Director of Machine Learning, Wikimedia Foundation

Benefits of Baseten for AI/ML infrastructure

It’s easy to get started with Baseten

At Baseten, we take AI/ML infrastructure seriously, and do everything we can to ensure you can get started in minutes, without getting tangled in complex deployment processes. Our model deployment features require zero infrastructure experience, and include logs and health metrics, giving you visibility into what’s happening with your models and making it easy to debug issues. And whether or not you’re a Baseten customer, you can always take advantage of Truss, our open-source model packaging framework.

Baseten performance and scalability

Our horizontally scalable infrastructure gets you from prototype to production as quickly as possible, and we couple that with blazingly fast inference times and autoscaling features that ensure all traffic to your model gets served, while giving you full control over scale to zero settings so that you’re not paying for idle GPUs. To get an idea of just how fast our cold starts are, for Stable Diffusion on an A10G, we reliably see cold start times of 15 seconds, from zero to ready for inference.

We also make sure that computing resource management is easy and straightforward so that you’re always clear on which GPUs you’re utilizing for a given model.

Learn more about model autoscaling features

Baseten cost-efficiency 

We enable a suite of customizable model infrastructure features, which, along with our GPU sharing, fast cold starts, and scale to zero functionality, mean that you only pay for the compute that you use. No more surprise bills for idle GPUs!

"If it’s just another brick in the infrastructure, don’t bother putting your most creative people on it. Just buy a solution that meets your needs and focus on whatever’s actually unique to your company."

Chris Albon

Baseten self-hosted deployment

Want to host models on your own cloud? Not a problem! Self-hosting comes with all the features of our Pro plan, and includes multi-tenant architecture capabilities, data segregation, and live engineering support. 

Baseten support 

Speaking of support, Baseten Pro and Self-hosted customers have dedicated engineering support through our team of forward-deployed engineers. These engineers are well-versed in the capabilities of Baseten and are deeply invested in ensuring your success with our platform.

Baseten security

We’re also proud of the fact that we’ve received both SOC II Type 2 certification as well as HIPAA compliance, demonstrating our dedication to providing secure and reliable infrastructure for hosting ML models, along with our commitment to ensuring the confidentiality, integrity, and availability of sensitive health information., 

Talk to us!

We’d love to know what you’re working on, and whether or not Baseten is a good fit for your project or startup. Reach out to us at–we’d love to hear from you!