---
license: mit
tags:
- physics
- pinn
- pde
- pytorch
- heat-equation
- burgers-equation
- schrodinger
model-index:
- name: Heat Equation 3D
  results:
  - task:
      type: physics-simulation
      name: Heat Equation
    dataset:
      type: synthetic
      name: Generated Data
    metrics:
    - type: mean_absolute_error
      name: Max Error
      value: 0.04
- name: Burgers' Equation 1D
  results:
  - task:
      type: physics-simulation
      name: Burgers' Equation
    dataset:
      type: synthetic
      name: Generated Data
    metrics:
    - type: mean_absolute_error
      name: Max Error
      value: 0.045
- name: Schrödinger Equation 1D
  results:
  - task:
      type: physics-simulation
      name: Schrödinger Equation
    dataset:
      type: synthetic
      name: Generated Data
    metrics:
    - type: mean_absolute_error
      name: Max Error
      value: 0.065
- name: Schrödinger Equation Hydrogen
  results:
  - task:
      type: physics-simulation
      name: Schrödinger Equation
    dataset:
      type: synthetic
      name: Generated Data
    metrics:
    - type: mean_absolute_error
      name: Max Error
      value: 0.02
---

# Physics-Informed Neural Networks
This contains the pretrained model weights for a collection of physics-informed neural networks (PINNs) that are designed to solve problems in quantum mechanics, thermodynamics, and fluid dynamics by learning directly from physics rather than from labeled data. For training and evaluation scripts, check out the [main repository](https://github.com/sr5434/pinns).

## Getting Started

### Prerequisites
- Python 3.12+
- PyTorch
- NumPy
- Matplotlib

### Installation
```bash
git clone https://github.com/sr5434/pinns.git
cd pinns
pip install -r requirements.txt
```

### Usage
#### Evaluate on Heat Equation
```bash
cd heat_equation

# Generate visualizations
curl "https://huggingface.co/sr5434/PINN-Collection/resolve/main/heat_equation_3d.pt?download=true" -o heat_equation_3d.pt
python heat_equation_visualizer_3d.py
```
#### Evaluate on Burger's Equation
```bash
cd burgers_equation

# Generate visualizations
curl "https://huggingface.co/sr5434/PINN-Collection/resolve/main/burgers_equation_1d.pt?download=true" -o burgers_equation_1d.pt
python burgers_equation_visualization_1d.py
```

#### Test Pretrained Model on Schrödinger Equation(1D)
```bash
cd schrodingers_equation

# Generate visualizations
curl -L "https://huggingface.co/sr5434/PINN-Collection/resolve/main/schrodingers_equation_1d.pt?download=true" -o schrodingers_equation_1d.pt
python schrodingers_visualization_1d.py
```


#### Test Pretrained Model on Schrödinger Equation(Hydrogen)
```bash
cd schrodingers_equation

# Generate visualizations
curl -L "https://huggingface.co/sr5434/PINN-Collection/resolve/main/schrodingers_equation_hydrogen.pt?download=true" -o schrodingers_equation_hydrogen.pt
python schrodingers_visualization_hydrogen.py
```