MLX How to use mlx-community/gemma-4-31B-it-assistant-bf16 with MLX:
# Make sure mlx-lm is installed
# pip install --upgrade mlx-lm
# if on a CUDA device, also pip install mlx[cuda]
# Generate text with mlx-lm
from mlx_lm import load, generate
model, tokenizer = load("mlx-community/gemma-4-31B-it-assistant-bf16")
prompt = "Once upon a time in"
text = generate(model, tokenizer, prompt=prompt, verbose=True) 
