examples/demo_custom_layer_backend_agnostic.py/MyDense/call
class MyDense:    def call(self, inputs):
        # Use Keras ops to create backend-agnostic layers/metrics/etc.
        return ops.matmul(inputs, self.w) + self.b