搜索

3 结果

清除过滤器
  • 2025年8月12日 / Kaggle

    在 TPU 上免费使用 JAX 训练 GPT2 模型

    在 Google TPU 上使用 JAX 从头开始构建和训练 GPT2 模型,并在免费版 Colab 或 Kaggle 中使用完整的 Python 笔记本。了解如何定义硬件网格、如何划分模型参数和输入数据以实现数据并行,以及如何优化模型训练流程。

    Train a GPT2 model with JAX on TPU for free
  • 2025年6月24日 / Kaggle

    结合使用 KerasHub 与 Hugging Face,轻松实现端到端机器学习工作流

    KerasHub 使用户能够混合搭配不同机器学习框架的模型架构和权重,支持将来自 Hugging Face Hub 等来源的检查点(包括使用 PyTorch 创建的检查点)加载到 Keras 模型中,并与 JAX、PyTorch 或 TensorFlow 配合使用。这种灵活性意味着,您可以充分利用大量经过社区微调的模型,同时完全掌控您所选择的后端框架。

    How to load model weights from SafeTensors into KerasHub for multi-framework machine learning
  • 2025年5月13日 / TensorFlow

    使用 Keras 和 JAX,在 10 分钟内即可完成 Recommender 系统的构建和训练

    Keras Recommenders (KerasRS) 是新推出的库,旨在帮助开发者使用带有排名和检索基本模块的 API 构建推荐系统。该库不仅可以通过 pip 安装,还支持 JAX、TensorFlow 或 PyTorch 后端。

    Build and train a Recommender System in 10 minutes using Keras and JAX