Haiku

This lesson will introduce Haiku, a high-level neural network library that provides an object-oriented interface.

Haiku is a tool

For building neural networks
Think: “Sonnet for JAX”

Introduction

If we check the official documentation of the Haiku, we will find the above literal haiku.

Like Keras or Sonnet for Tensorflow, Haiku is a high-level library that provides an object-oriented interface for building neural networks (in JAX).

As a starter, let’s use a two-layer MLP.

Get hands-on with 1400+ tech skills courses.