3 ms·

Any chance a keras.linalg suite of ops is in the works? Cross-platform differences between the behavior of tf.linalg and torch.linalg have cost me a lot of tim

Any chance a keras.linalg suite of ops is in the works? Cross-platform differences between the behavior of tf.linalg and torch.linalg have cost me a lot of time over the years.

kerasteam 3y ago
We don't have a separate `ops.linalg` package, but we do include `numpy.linalg` ops as part of `keras.ops`. For now only 2 ops are supported: `qr` and `solve`. We're open to adding any `numpy.linalg` op that turns out to be useful (or you could open a PR for any op you need).