rename from stip to stix

README.md

@@ -1,6 +1,6 @@
 ---
 license: apache-2.0
-library_name: stip
+library_name: stix
 tags:
 - jax
 - flax
@@ -8,16 +8,16 @@ tags:
 - tutorial
 ---
 
-# STIP tutorial checkpoints
+# STIX tutorial checkpoints
 
-Small checkpoints used by the [`stip`](https://github.com/instadeepai/stip) tutorial
+Small checkpoints used by the [`stix`](https://github.com/instadeepai/stix) tutorial
 notebooks, so that a tutorial can demonstrate sampling without spending ten
 minutes training first. They are toy models (a two-layer MLP, 18k-25k parameters,
 trained for 3000 steps on a 4-component 2D Gaussian mixture) and have no
 use outside the notebooks.
 
 Checkpoints are [Orbax](https://orbax.readthedocs.io) directories written by
-`stip`'s own `TrainingIOHandler`, holding `params`, `opt_state`, `ema_params` and
+`stix`'s own `TrainingIOHandler`, holding `params`, `opt_state`, `ema_params` and
 `extra` (EMA decay and step count) as separately-restorable items.
 
 ## `conditioning_and_guidance/`
@@ -36,10 +36,10 @@ different modalities:
 ```python
 from flax import nnx
 from huggingface_hub import snapshot_download
-from stip.training.checkpointer import Checkpointer, CheckpointerConfig
+from stix.training.checkpointer import Checkpointer, CheckpointerConfig
 
 path = snapshot_download(
-    "InstaDeepAI/STIP-tutorials", allow_patterns="conditioning_and_guidance/joint_model/*"
+    "InstaDeepAI/STIX-tutorials", allow_patterns="conditioning_and_guidance/joint_model/*"
 )
 gen_model = ...  # build the same model structure as the notebook
 graphdef, params = nnx.split(gen_model, nnx.Param)