Developing a QNN or QCNN using Flax

[Gopal-Dahale]

Aim

Developing a Quantum neural network or convolutional neural network using Flax

Details

The idea is to first create a simple quantum neural network (the simplest being a single qubit classifier) in Flax. If possible build upon it to create a QCNN. A tutorial is present on how to create QCNN in JAX. This tutorial is based on the neural network tutorial from JAX and Using JAX with PennyLane.

In JAX, the user has to define the classical neural networks from scratch and therefore we want to use Flax to make this easier.

Resources

• See the QCNN JAX tutorial to understand how the quantum model is built in JAX.
• Read the Flax documentation on managing parameters and state.