Training Tensorflow Models using Airflow

Airflow

Apache Airflow is an open-source platform for authoring, scheduling and monitoring data and computing workflows. It was first developed by Airbnb and is now under the Apache Software Foundation. Airflow uses Python to create workflows that can be easily scheduled and monitored. Airflow can help you move data from one source to a destination, filter datasets, apply data policies, manipulation, monitoring and even call microservices to trigger database management tasks. It can be used for batch jobs, organizing, monitoring, and executing workflows automatically. Airflow has been used by many companies for various use cases such as ETL pipelines, machine learning workflows, data warehousing, and more.

Tensorflow Models

TensorFlow is a free and open-source software library for machine learning and artificial intelligence. It can be used across a range of tasks but has a particular focus on training and inference of deep neural networks. TensorFlow helped launch the deep learning revolution: with pre-trained models, data, and high-level APIs, it has become easy for everyone to build ML and DL models. One of the important Tensorflow features is that it is flexible in its operating model, meaning it has modularity and the parts of it that you want to make standalone it offers you that option. TensorFlow.js is a JavaScript-based framework to run machine learning models within the browser. Any modern browser can run the TensorFlow model with no changes to the code.

Open source orchestrators like Airflow are one of the primary means by which companies train Tensorflow models in production. Airflow offers a mechanism to schedule and monitor these jobs as part of more complex workflow graphs. Kaspian has a native operator for Airflow; this operator makes it easy to either swap to or get started with training pipelines that utilize Kaspian's flexible compute layer, with native support for autoscaling, GPU acceleration, and more.

Learn more about Kaspian and see how our flexible compute layer for the modern data cloud is already reshaping the way companies in industries like retail, manufacturing and logistics are thinking about data engineering and analytics.