Views
No views yet
AI-Powered Physics Simulation & Mathematical Animation GenerationExplicit Spatial Reasoning + API-Centric Code Generation
Askit-OLMo-32B Model
↓
Generates Code
↓
PhysicsBridge API
↓
Askit. Platform
↓
Real-time Visualization| Project | Purpose | Link |
|---|---|---|
| Askit. | Interactive Animation Platform | GitHub |
| PhysicsBridge | Physics Engine Wrapper | Integrated in Askit. |
| OLMo-3.1-32B | Base Model | Allen AI |
| 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 |
<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>1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_id = "SStarrySSky/Askit-OLMo-32B-Spatial-Thinking-Preview"
4tokenizer = AutoTokenizer.from_pretrained(model_id)
5model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto")
6
7# Physics simulation with spatial reasoning
8prompt = """
9Create a physics simulation for a ball dropped from 10 meters.
10Ball mass: 1kg, initial velocity: (0, 0, 0)
11Use PhysicsBridge API.
12"""
13
14inputs = tokenizer(prompt, return_tensors="pt")
15outputs = model.generate(**inputs, max_length=2048, temperature=0.7)
16print(tokenizer.decode(outputs[0]))