Understanding PyTorch Ax

Developed by Meta, Ax is one of the most flexible frameworks for A/B testing in deep learning models.

Image Credit: Meta AI

I recently started an AI-focused educational newsletter, that already has over 100,000 subscribers. TheSequence is a no-BS (meaning no hype, no news etc) ML-oriented newsletter that takes 5 minutes to read. The goal is to keep you up to date with machine learning projects, research papers and concepts. Please give it a try by subscribing below:

TheSequence

A/B testing is one of the most important elements of machin learning(ML) pipelines. In the ML ecosystem, PyTorch has become one of the favorite frameworks for implemting experiments in ML models. Recently, Meta(Facebook) AI open sourced Ax, a new framework for rapid A/B testing and experimentation in ML models.

Conceptually, Ax is a platform for optimizing experiments such as A/B tests, simulations or machine learning models. Ax is dependent on BoTorch, another recently released PyTorch framework. Ax provides a high-level, easy-to-use API to interface with BoTorch allowing…