πŸš€ 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.

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

πŸ› οΈ Ecosystem & Integration

πŸ“± Official Software

πŸ’» GitHub: github.com/SwayingWheatfield/Askit.

πŸ”— 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

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

Downloads last month
44
Safetensors
Model size
32B params
Tensor type
BF16
Β·
Inference Providers NEW
This model isn't deployed by any Inference Provider. πŸ™‹ Ask for provider support

Evaluation results

  • accuracy on CPhO & IMO Level Problems
    self-reported
    High spatial reasoning accuracy