Is PSA mbed-crypto (using secure element) usable with TLSSocket / TLSSocketWrapper?

I got PSA with secure element functionality implemented using this repo as a reference:

According to this document:

I need to replace mbedtls_pk_parse_key() and mbedtls_ssl_conf_own_cert()
these are present in the TLSSocketWrapper class in the set_client_cert_key method.

Is it possible to implement TLSSocket functionality with PSA or do I need to patch this in myself or are there any future plans to implement this?

Any other suggestions on how to approach this are also welcome.

I have heard that the TLSSocketWrapper has no PSA implementation yet. I’m gonna try using psa & mbed TLS.