keras-nlp
KerasHub provides a comprehensive library that supports natural language processing, computer vision, audio, and multimodal models on TensorFlow, JAX, and PyTorch. Developed using Keras 3, it features a rich collection of pre-trained models and foundational components suitable for diverse applications. The library ensures consistent model definitions across frameworks, facilitating straightforward fine-tuning on both GPUs and TPUs. KerasHub enhances performance with support for model and data parallel training, offering seamless model migrations without additional costs across platforms.
