Add simplified APIs for model obtaining maxtext models#3450

A9isha wants to merge 1 commit intomainAI-Hypercomputer/maxtext:mainfrom

anisha-from-pretrained3AI-Hypercomputer/maxtext:anisha-from-pretrained3Copy head branch name to clipboard