Update flax in GPU environments (#1541)

tifftoff · web-flow · commit e788e63226cd · 2026-03-17T10:54:08.000-04:00
b/493548400 https://www.kaggle.com/discussions/product-feedback/680483
diff --git a/kaggle_requirements.txt b/kaggle_requirements.txt
@@ -29,6 +29,7 @@ emoji
 fasttext
 featuretools
 fiona
+flax
 fury
 fuzzywuzzy
 geojson
