最近LLMを使った実験をする機会があったので,その実験管理に最新バージョン(2.9.2)のmlflowを使ってみた.そのバージョンではLLMに対するいくつかの便利機能があったので,今回はそれについて書いてみる. Mlflow MLflowは、機械学習のライフサイクルを管理するためのOSSで,モデルのバージョニング、データセットの管理、実験の追跡、デプロイメントなど、機械学習プロジェクトの全体的なプロセスをサポートしている. ドキュメントも充実しており,Mlflowの使い方というより,機械学習の実験管理のベストプラクティスを学ぶ際にもとても有用なものだと思う. MlflowのLLM用機能 Mlflowの最新バージョン(2.9.2)では,LLMの実験管理用の機能がいくつか追加されている. 内容は大きく分けて以下の3つで,それぞれについて紹介していく. デプロイメント ロギング(トラッキング)