Attach probe heads to a language model and train them jointly with the base model, then read their predictions live during generation. One API across Apple MLX and PyTorch.
-
Updated
Sep 16, 2026 - Python
#
joint-training
Here is 1 public repository matching this topic...
Add this topic to your repo
To associate your repository with the joint-training topic, visit your repo's landing page and select "manage topics."