Amsi-fin-o1 MLX (BFloat16)

Model Logo

Financial Vision-Language Model optimized for Apple Silicon

MLX Apple Silicon License

Model Description

This is the BFloat16 MLX conversion of AITRADER/Amsi-fin-o1, a specialized financial vision-language model designed for analyzing financial documents, charts, and performing complex financial reasoning.

Key Features

  • Financial Document Analysis: Extract and analyze information from financial statements, reports, and documents
  • Chart Understanding: Interpret financial charts, graphs, and visualizations with high accuracy
  • Chain-of-Thought Reasoning: Advanced thinking capabilities for complex financial calculations and analysis
  • Multi-Modal Input: Process both images and text for comprehensive financial analysis
  • Apple Silicon Optimized: Native performance on M1/M2/M3/M4 chips

Model Architecture

Component Specification
Base Architecture Qwen3-VL (4B parameters)
Text Model 36 layers, 2560 hidden size
Vision Encoder 24 layers, 1024 hidden size
Attention Heads 32 (8 KV heads)
Context Length Up to 131,072 tokens
Precision BFloat16
Model Size ~8GB

Installation

# Install mlx-vlm
pip install -U mlx-vlm

Quick Start

Command Line

# Basic image analysis
python -m mlx_vlm.generate \
    --model AITRADER/Amsi-fin-o1-MLX-bf16 \
    --max-tokens 512 \
    --temperature 0.7 \
    --prompt "Analyze this financial chart and explain the trends." \
    --image path/to/financial_chart.png

Python API

from mlx_vlm import load, generate
from mlx_vlm.prompt_utils import apply_chat_template
from mlx_vlm.utils import load_config

# Load model
model_path = "AITRADER/Amsi-fin-o1-MLX-bf16"
model, processor = load(model_path)
config = load_config(model_path)

# Prepare prompt
prompt = apply_chat_template(
    processor,
    config,
    "Analyze the financial performance shown in this quarterly report.",
    num_images=1
)

# Generate response
output = generate(
    model,
    processor,
    prompt,
    image="path/to/report.png",
    max_tokens=512,
    temperature=0.7
)
print(output)

Streaming Generation

from mlx_vlm import load, stream_generate
from mlx_vlm.prompt_utils import apply_chat_template
from mlx_vlm.utils import load_config

model_path = "AITRADER/Amsi-fin-o1-MLX-bf16"
model, processor = load(model_path)
config = load_config(model_path)

prompt = apply_chat_template(
    processor,
    config,
    "What are the key insights from this financial statement?",
    num_images=1
)

# Stream the response
for token in stream_generate(
    model,
    processor,
    prompt,
    image="financial_statement.png",
    max_tokens=512
):
    print(token, end="", flush=True)

Use Cases

1. Financial Chart Analysis

prompt = """Analyze this stock chart and provide:
1. The overall trend (bullish/bearish/neutral)
2. Key support and resistance levels
3. Volume analysis
4. Trading recommendations"""

2. Financial Statement Review

prompt = """Review this income statement and:
1. Calculate key financial ratios
2. Compare YoY performance
3. Identify areas of concern
4. Highlight positive indicators"""

3. Document OCR and Extraction

prompt = """Extract all numerical data from this financial document
and organize it in a structured format."""

4. Investment Analysis

prompt = """Based on this quarterly report:
1. Summarize the company's financial health
2. Calculate growth metrics
3. Provide an investment thesis"""

Performance Benchmarks

Benchmark Score
Financial Document OCR High Accuracy
Chart Interpretation Excellent
Numerical Reasoning Strong
Multi-step Analysis Advanced

Hardware Requirements

Apple Silicon Performance
M1 (8GB) Functional (may require swap)
M1 Pro/Max (16GB+) Good
M2/M2 Pro/Max Very Good
M3/M3 Pro/Max Excellent
M4/M4 Pro/Max Best

Recommended: 16GB+ unified memory for optimal performance

Model Variants

Variant Precision Size Speed Quality
bf16 BFloat16 ~8GB Baseline Highest
8bit 8-bit Quantized ~4GB Faster Very Good

Training Data

This model was fine-tuned on specialized financial datasets:

  • FinTrain: Comprehensive financial training data
  • MultiFinBen-EnglishOCR: Multi-lingual financial document OCR
  • SecureFinAI Contest: Financial security and analysis tasks
  • ChartQA: Chart question answering
  • NuminaMath-CoT: Mathematical reasoning with chain-of-thought
  • FinCoT: Financial chain-of-thought reasoning
  • COTA-LLaVA: Complex reasoning visual data

Limitations

  • Primarily trained on English financial data
  • May require prompt engineering for optimal results
  • Performance varies with image quality
  • Not a replacement for professional financial advice

Citation

@misc{amsi-fin-o1-mlx,
  title={Amsi-fin-o1 MLX: Financial Vision-Language Model for Apple Silicon},
  author={AITRADER},
  year={2025},
  url={https://huggingface.co/AITRADER/Amsi-fin-o1-MLX-bf16}
}

Acknowledgments

License

This model is released under the Apache 2.0 License.


Optimized for Apple Silicon with MLX
Made with Apple MLX Framework
Downloads last month
18
Safetensors
Model size
4B params
Tensor type
BF16
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for AITRADER/Amsi-fin-o1-MLX-bf16

Datasets used to train AITRADER/Amsi-fin-o1-MLX-bf16