Kernel Died Importing Keras on Mac M1

Hi,

I am trying to import keras as followed on my Mac M1 running on MacOS 11.6 but the kernel died during the process.

from keras.models import Sequential from keras.layers import Dense

The error message in the terminal: [I 15:44:39.003 NotebookApp] Kernel started: e0bec9a2-cde8-42ed-be9d-747ea2841818, name: python3

2022-01-27 15:44:43.203092: F tensorflow/c/c_api_experimental.cc:739] Non-OK-status: tensorflow::RegisterPluggableDevicePlugin(lib_handle->lib_handle) status: FAILED_PRECONDITION: 'host_callback' field in SP_StreamExecutor must be set.

Not sure if it's related, I have read the documentation on https://developer.apple.com/metal/tensorflow-plugin/ and the OS requirement is MacOS 12.0+.

Is there anyway to solve this issue on my current OS?