add flax framework and its safetensors backend to JSON schema examples

CHANGELOG.md

@@ -9,8 +9,10 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ### Added
 - Add [`huggingface/safetensors`](https://github.com/huggingface/safetensors)
-  recommendations for ``mlm:artifact_type`` and corresponding ``mlm:framework`` values
+  recommendations for `mlm:artifact_type` and corresponding ``mlm:framework`` values
   (fixes [#68](https://github.com/stac-extensions/mlm/issues/68)).
+- Add [`Flax`](https://github.com/google/flax) to the list of `mlm:framework` and
+  the corresponding `mlm:artifact_type` SafeTensors backend in the JSON schema examples.
 - Add [`Paddle`](https://github.com/PaddlePaddle/Paddle) to the list of `mlm:framework`
   (fixes [#69](https://github.com/stac-extensions/mlm/issues/69)).
 

README.md

@@ -238,6 +238,7 @@ to use common names when applicable. Below are a few notable entries.
 - [`rgee`](https://github.com/r-spatial/rgee)
 - [`spatialRF`](https://github.com/BlasBenito/spatialRF)
 - [`JAX`](https://github.com/jax-ml/jax)
+- [`Flax`](https://github.com/google/flax)
 - [`MXNet`](https://github.com/apache/mxnet)
 - [`Caffe`](https://github.com/BVLC/caffe)
 - [`PyMC`](https://github.com/pymc-devs/pymc)

json-schema/schema.json

@@ -436,6 +436,7 @@
             "rgee",
             "spatialRF",
             "JAX",
+            "Flax",
             "MXNet",
             "Caffe",
             "PyMC",
@@ -468,6 +469,7 @@
         "tf.keras.Model.export",
         "safetensors.torch.save_file",
         "safetensors.tensorflow.save_file",
+        "safetensors.flax.save_file",
         "safetensors.paddle.save_file"
       ]
     },