NeuralTxt Reward Model MLX

MLX-ready package for paperbd/neuraltxt-reward-tiny.

This repo contains:

  • model.safetensors: MiniLM/BERT encoder weights readable by MLX.
  • reward_head.safetensors: clamped linear reward head in MLX safetensors format.
  • tokenizer files copied from the source reward model.

Usage

from neuraltxt import NeuralTxtReward

reward = NeuralTxtReward(backend="mlx")
score = reward.score(
    response="Paris is the capital of France.",
    reference="The capital of France is Paris.",
)

To load this repo explicitly:

reward = NeuralTxtReward("paperbd/neuraltxt-reward-tiny-mlx", backend="mlx")
Downloads last month
79
Safetensors
Model size
22.7M params
Tensor type
F32
·
MLX
Hardware compatibility
Log In to add your hardware

Quantized

Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for paperbd/neuraltxt-reward-tiny-mlx