refactor(framework) Update `mlx`,`tf` and `pytorch` templates #13669

Job	Run time
Build, test and upload wheel	1m 33s
Framework / bare	3m 33s
Strategy / FedMedian	1m 15s
Framework / bare-https	1m 25s
Template / numpy	47s
Strategy / FedTrimmedAvg	1m 37s
Framework / bare-client-auth	53s
Template / pytorch	2m 31s
Strategy / QFedAvg	1m 15s
Framework / framework-jax	1m 21s
Template / tensorflow	2m 58s
Framework / framework-pytorch	2m 9s
Strategy / FaultTolerantFedAvg	1m 10s
Template / jax	1m 4s
Strategy / FedAvgM	1m 13s
Framework / framework-tensorflow	3m 16s
Template / sklearn	2m 17s
Strategy / FedAdam	1m 8s
Framework / framework-opacus	2m 17s
Strategy / FedAdagrad	1m 10s
Framework / framework-pytorch-lightning	1m 38s
Strategy / FedYogi	1m 13s
Framework / framework-scikit-learn	1m 43s
Framework / framework-fastai	2m 51s
Framework / framework-pandas	1m 12s
	43m 29s

Provide feedback