検索

5 結果

フィルタをクリアする
  • 2025年8月12日 / Kaggle

    JAX を使用して TPU で GPT2 モデルを無料でトレーニング

    Google TPU で JAX を使用し、GPT2 モデルをゼロから構築してトレーニングしましょう。これには、無料枠の Colab または Kaggle で利用できる完全な Python ノートブックを使用します。ハードウェア メッシュの定義、モデル パラメータと入力データのデータ並列処理のための分割、モデル トレーニング プロセスの最適化について学びましょう。

    Train a GPT2 model with JAX on TPU for free
  • 2025年7月29日 / AI

    JAX を使用したロボット工学者の取り組み: 最適制御とシミュレーションにおける効率性の追求

    Max 氏の体験談では、JAX ネイティブの LQR ソルバーである LQRax を紹介します。Brax、MJX、JaxSim などのツールを含む、成長し続ける JAX ロボット工学エコシステムを例示し、最適制御とシミュレーションにおける計算効率や、モデルベースと学習ベースのアプローチのシームレスな統合といった JAX の利点に注目していきます。

    JAX_meta
  • 2025年7月16日 / Cloud

    スタンフォード大学の Marin 基盤モデル: JAX を使用して開発された初の完全オープンモデル

    Marin プロジェクトは、開発プロセス全体をアクセス可能で再現可能なものにすることで、AI における「オープン」の定義をモデル自体にとどまらず科学的プロセス全体にまで拡張することを目指しています。この取り組みは JAX フレームワークとその Levanter ツールを活用しており、基盤モデルの詳細な精査、信頼性の確保、そのうえでの構築を可能にすることで、より透明性の高い AI 研究の未来を育みます。

    Stanford Marin project in JAX
  • 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 分でレコメンデーション システムを作ってトレーニングする

    Keras Recommenders(KerasRS)は、レコメンデーション システムを開発するための新しいライブラリです。pip でインストールでき、ランキングや検索のビルディング ブロックとなる API を利用できます。JAX、TensorFlow、PyTorch バックエンドがサポートされています。

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