JaxBackend.size ============================ .. currentmodule:: tme.backends .. automethod:: JaxBackend.size