|
|
--- |
|
|
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](https://askit.space) |
|
|
- ๐ป **GitHub**: [github.com/SStarrySSky/Askit.](https://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](https://github.com/SStarrySSky/Askit.) | |
|
|
| **PhysicsBridge** | Physics Engine Wrapper | Integrated in Askit. | |
|
|
| **OLMo-3.1-32B** | Base Model | [Allen AI](https://huggingface.co/allenai/OLMo-3.1-32B-Instruct) | |
|
|
|
|
|
--- |
|
|
|
|
|
## ๐ 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 |
|
|
|
|
|
```python |
|
|
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 |
|
|
- ๐ **Website**: [askit.space](https://askit.space) |
|
|
- ๐ป **GitHub Repository**: [github.com/SStarrySSky/Askit.](https://github.com/SStarrySSky/Askit.) |
|
|
- ๐ค **HuggingFace Model**: [SStarrySSky/Askit-OLMo-32B-Spatial-Thinking-Preview](https://huggingface.co/SStarrySSky/Askit-OLMo-32B-Spatial-Thinking-Preview) |
|
|
|
|
|
### Base Technologies |
|
|
- ๐ง **OLMo-3.1-32B**: [allenai/OLMo-3.1-32B-Instruct](https://huggingface.co/allenai/OLMo-3.1-32B-Instruct) |
|
|
- โ๏ธ **Transformers**: [huggingface.co/transformers](https://huggingface.co/transformers/) |
|
|
|
|
|
--- |
|
|
|
|
|
## ๐ License |
|
|
|
|
|
**GPL-3.0 License** - See [LICENSE](https://github.com/SStarrySSky/Askit./blob/main/LICENSE) |
|
|
|
|
|
--- |
|
|
|
|
|
## ๐ Acknowledgments |
|
|
|
|
|
Built on top of **OLMo-3.1-32B-Instruct** by Allen Institute for AI. |
|
|
|
|
|
Integrated with **Askit.** - Interactive Physics Animation Platform. |
|
|
|
|
|
--- |
|
|
|
|
|
<div align="center"> |
|
|
|
|
|
**Made with โค๏ธ by Starry Sky** |
|
|
|
|
|
[๐ Website](https://askit.space) โข [๐ป GitHub](https://github.com/SStarrySSky/Askit.) โข [๐ค HuggingFace](https://huggingface.co/SStarrySSky/Askit-OLMo-32B-Spatial-Thinking-Preview) |
|
|
|
|
|
</div> |
|
|
|