For what it's worth, here's what's logged:
2024-07-21 13:02:39.605814: I metal_plugin/src/device/metal_device.cc:1154] Metal device set to: Apple M2 Max
2024-07-21 13:02:39.605835: I metal_plugin/src/device/metal_device.cc:296] systemMemory: 32.00 GB
2024-07-21 13:02:39.605840: I metal_plugin/src/device/metal_device.cc:313] maxCacheSize: 10.67 GB
2024-07-21 13:02:39.605869: I tensorflow/core/common_runtime/pluggable_device/pluggable_device_factory.cc:306] Could not identify NUMA node of platform GPU ID 0, defaulting to 0. Your kernel may not have been built with NUMA support.
2024-07-21 13:02:39.605883: I tensorflow/core/common_runtime/pluggable_device/pluggable_device_factory.cc:272] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 0 MB memory) -> physical PluggableDevice (device: 0, name: METAL, pci bus id: <undefined>)