SStarrySSky's picture
Rollback to enhanced version with YAML metadata
7dd5fd9 verified
metadata
language:
  - en
  - zh
license: gpl-3.0
tags:
  - physics
  - mathematics
  - spatial-reasoning
  - code-generation
  - ollama
  - gguf
  - lora
  - deepspeed
  - qlora
  - fine-tuning
datasets:
  - physics-problems
  - mathematics-problems
model-index:
  - name: Askit-OLMo-32B-Spatial-Thinking-Preview
    results:
      - task:
          type: text-generation
        dataset:
          type: physics-math-problems
          name: CPhO & IMO Level Problems
        metrics:
          - type: accuracy
            value: High spatial reasoning accuracy

๐Ÿš€ Askit-OLMo-32B-Spatial-Thinking-Preview

AI-Powered Physics Simulation & Mathematical Animation Generation

Explicit Spatial Reasoning + API-Centric Code Generation


โœจ What Makes Askit Special?

Askit-OLMo-32B-Spatial-Thinking-Preview is not just another code generation model. It's a spatial reasoning specialist that thinks in 3D coordinates before writing code.

๐Ÿง  Core Innovation: Forced Spatial Cognition

Unlike traditional models that generate code directly, Askit enforces a complete spatial thinking pipeline:

Physics Problem
    โ†“
ใ€็ฉบ้—ด็›ด่ง‰ๅˆ†ๆžใ€‘ (3D Space Analysis)
    โ†“
ใ€็ฉบ้—ดๅๆ ‡่ฎก็ฎ—ใ€‘ (Coordinate Calculations)
    โ†“
ใ€็‰ฉ็†โ†’API็ฟป่ฏ‘ใ€‘ (Physics โ†’ API Translation)
    โ†“
Production-Ready Code

Key Features:

  • โœ… Forced Spatial Cognition: Every problem decomposed into 3D coordinates
  • โœ… API-Centric Reasoning: Physics principles โ†’ PhysicsBridge API calls
  • โœ… Explicit Reasoning: Complete thinking chain visible in output
  • โœ… Competition-Ready: Optimized for CPhO & IMO level problems

๐ŸŽฏ Capabilities

1. Spatial Understanding ๐Ÿ—บ๏ธ

  • 3D object position relationships
  • Precise coordinate calculations (x, y, z)
  • Automatic trajectory derivation
  • Explicit spatial reasoning in every response

2. Physics Modeling โš™๏ธ

  • Classical mechanics, electromagnetism, thermodynamics
  • API-Driven: Direct PhysicsBridge API generation
  • Rigid body collisions, SPH fluids, ODE/PDE solvers
  • Complex multi-body systems with constraints

3. Advanced Problem Solving ๐Ÿ†

  • CPhO (Chinese Physics Olympiad) level optimization
  • IMO (International Mathematical Olympiad) level optimization
  • Deep reasoning for competition problems
  • Production-ready simulation code

๐Ÿ› ๏ธ Ecosystem & Integration

๐Ÿ“ฑ Official Software

Askit. - Interactive Physics Animation Platform

  • ๐ŸŒ Website: askit.space
  • ๐Ÿ’ป GitHub: github.com/SStarrySSky/Askit.
  • ๐Ÿ“ฆ Features:
    • Real-time physics simulation with OpenGL rendering
    • Interactive controls (sliders, buttons, checkboxes)
    • Timeline-based animation control
    • Snapshot system for AI context awareness
    • PhysicsBridge API integration

๐Ÿ”— Integration Points

Askit-OLMo-32B Model
        โ†“
    Generates Code
        โ†“
PhysicsBridge API
        โ†“
Askit. Platform
        โ†“
Real-time Visualization

๐Ÿ“š Related Projects

Project Purpose Link
Askit. Interactive Animation Platform GitHub
PhysicsBridge Physics Engine Wrapper Integrated in Askit.
OLMo-3.1-32B Base Model Allen AI

๐Ÿ“Š Model Specifications

Aspect Details
Base Model OLMo-3.1-32B-Instruct
Fine-tuning LoRA (Rank 256)
Training Data 3,500+ physics/math problems
Framework DeepSpeed ZeRO-3 + BF16
Hardware 3x RTX 5090 GPUs
Output Format Explicit reasoning chains + code

๐Ÿ’ก Output Format

The model generates complete reasoning chains with explicit spatial thinking:

<thought>
ใ€็ฉบ้—ด็›ด่ง‰ๅˆ†ๆžใ€‘
- 3D space structure analysis
- Initial positions (xโ‚€, yโ‚€, zโ‚€)
- Initial velocities (vโ‚“, vแตง, vแตค)
- Coordinate system setup

ใ€็‰ฉ็†ๅŽŸ็†ๆŽจๅฏผใ€‘
- Applicable physics laws
- Force analysis
- Acceleration calculations

ใ€็ฉบ้—ดๅๆ ‡่ฎก็ฎ—ใ€‘
- Position at time t: (x(t), y(t), z(t))
- Velocity vector: (vโ‚“(t), vแตง(t), vแตค(t))
- Trajectory equations

ใ€็‰ฉ็†โ†’API็ฟป่ฏ‘ใ€‘
- PhysicsBridge API calls
- Parameter mapping: coordinates โ†’ API
- Initial conditions setup
</thought>

<code>
# PhysicsBridge API Integration
physics = PhysicsBridge()
physics.create_rigid_body(
    position=(xโ‚€, yโ‚€, zโ‚€),
    velocity=(vโ‚“, vแตง, vแตค),
    mass=m,
    shape='sphere'
)
# ... more API calls
</code>

๐Ÿš€ Quick Start

from transformers import AutoModelForCausalLM, AutoTokenizer

model_id = "SStarrySSky/Askit-OLMo-32B-Spatial-Thinking-Preview"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto")

# Physics simulation with spatial reasoning
prompt = """
Create a physics simulation for a ball dropped from 10 meters.
Ball mass: 1kg, initial velocity: (0, 0, 0)
Use PhysicsBridge API.
"""

inputs = tokenizer(prompt, return_tensors="pt")
outputs = model.generate(**inputs, max_length=2048, temperature=0.7)
print(tokenizer.decode(outputs[0]))

๐Ÿ“– Use Cases

๐ŸŽ“ Physics Education

  • Interactive animations for teaching concepts
  • Explicit spatial reasoning aids student understanding
  • API-driven code runs directly in Askit. platform

๐Ÿ“Š Mathematical Visualization

  • Visual demonstrations of math problems
  • Geometric accuracy through coordinate calculations
  • Perfect for IMO-level problem visualization

๐Ÿ”ฌ Research Simulation

  • Academic research physics simulations
  • Correct coordinate systems guaranteed
  • Real-time rendering via PhysicsBridge

๐Ÿ† Competitive Problem Solving

  • CPhO and IMO level problem solving
  • Forced spatial reasoning matches competition requirements
  • Production-ready simulation code

๐Ÿ”— Links & Resources

Official Channels

Base Technologies


๐Ÿ“„ License

GPL-3.0 License - See LICENSE


๐Ÿ™ Acknowledgments

Built on top of OLMo-3.1-32B-Instruct by Allen Institute for AI.

Integrated with Askit. - Interactive Physics Animation Platform.


Made with โค๏ธ by Starry Sky

๐ŸŒ Website โ€ข ๐Ÿ’ป GitHub โ€ข ๐Ÿค— HuggingFace