model.onnx

ONNX Model Export
Generated: 2025-12-28T17:48:01.525Z
Model: micro-distill-grpo-vae
Framework: PyTorch 2.0+
ONNX Version: 1.14.0

Graph Structure:
- Inputs:
  - input_ids: int64[batch_size, sequence_length]
  - attention_mask: int64[batch_size, sequence_length]
  - position_ids: int64[batch_size, sequence_length] (optional)

- Outputs:
  - logits: float32[batch_size, sequence_length, 50257]

Layers:
  - Embedding: 512 dimensions
  - 8 Transformer blocks
  - Layer normalization
  - Language modeling head

Optimizations:
  - Constant folding: enabled
  - Shape inference: enabled
  - Dynamic axes: sequence_length
  - Opset: 17
  - IR version: 9

Quantization: Q4 (4-bit)