Skip to main content

A domain-specific language and debugger for neural networks

Project description

Design sans titre (2)

⚠️ WARNING: Neural-dsl is a WIP DSL and debugger—bugs exist, feedback welcome! This project is under active development and not yet production-ready!

Neural: A Neural Network Programming Language

License: MIT Python 3.8+ Discord Pylint Python package CodeQL Advanced Tests Coverage Dev.to blog Medium PyPi

design-01jmphv5f1-1740433387

Neural is a domain-specific language (DSL) designed for defining, training, debugging, and deploying neural networks. With declarative syntax, cross-framework support, and built-in execution tracing (NeuralDbg), it simplifies deep learning development.

🚀 Features

  • YAML-like Syntax: Define models intuitively without framework boilerplate.
  • Shape Propagation: Catch dimension mismatches before runtime.
    • ✅ Interactive shape flow diagrams included.
  • Multi-Framework HPO: Optimize hyperparameters for both PyTorch and TensorFlow with a single DSL config (#434).
  • Multi-Backend Export: Generate code for TensorFlow, PyTorch, or ONNX.
  • Training Orchestration: Configure optimizers, schedulers, and metrics in one place.
  • Visual Debugging: Render interactive 3D architecture diagrams.
  • Extensible: Add custom layers/losses via Python plugins.

🛠 NeuralDbg: Built-in Neural Network Debugger

NeuralDbg provides real-time execution tracing, profiling, and debugging, allowing you to visualize and analyze deep learning models in action.

Real-Time Execution Monitoring – Track activations, gradients, memory usage, and FLOPs.
test_trace_graph test_flops_memory_chart test_trace_graph_stacked test_trace_graph_heatmap test_anomaly_chart test_dead_neurons test_gradient_chart

Shape Propagation Debugging – Visualize tensor transformations at each layer.
Gradient Flow Analysis – Detect vanishing & exploding gradients.
Dead Neuron Detection – Identify inactive neurons in deep networks.
Anomaly Detection – Spot NaNs, extreme activations, and weight explosions.
Step Debugging Mode – Pause execution and inspect tensors manually.

📦 Installation

Clone the repository

git clone https://github.com/yourusername/neural.git cd neural

Create a virtual environment (recommended)

python -m venv venv source venv/bin/activate # Linux/macOS venv\Scripts\activate # Windows

Install dependencies

pip install -r requirements.txt
pip install neural-dsl

see v0.2.3 for bug fixes

Prerequisites: Python 3.8+, pip

🛠️ Quick Start

1. Define a Model

Create mnist.neural:

network MNISTClassifier {
  input: (28, 28, 1)  # Channels-last format
  layers:
    Conv2D(filters=32, kernel_size=(3,3), activation="relu")
    MaxPooling2D(pool_size=(2,2))
    Flatten()
    Dense(units=128, activation="relu")
    Dropout(rate=0.5)
    Output(units=10, activation="softmax")
  
  loss: "sparse_categorical_crossentropy"
  optimizer: Adam(learning_rate=0.001)
  metrics: ["accuracy"]
  
  train {
    epochs: 15
    batch_size: 64
    validation_split: 0.2
  }
}

3. Run Or Compile The Model

neural run mnist.neural --backend tensorflow --output mnist_tf.py
# Or for PyTorch:
neural run mnist.neural --backend pytorch --output mnist_torch.py

4. Visualize Architecture

neural visualize mnist.neural --format png

This will create architecture.png, shape_propagation.html, and tensor_flow.html for inspecting the network structure and shape propagation.

MNIST Architecture

5. Debug with NeuralDbg

neural debug mnist.neural

Open your browser to http://localhost:8050 to monitor execution traces, gradients, and anomalies interactively.

6. Use The No-Code Interface

neural --no_code

Open your browser to http://localhost:8051 to build and compile models via a graphical interface.


🛠 Debugging with NeuralDbg

🔹 1️⃣ Start Real-Time Execution Tracing

python neural.py debug mnist.neural

Features:
✅ Layer-wise execution trace
✅ Memory & FLOP profiling
✅ Live performance monitoring

🔹 2️⃣ Analyze Gradient Flow

python neural.py debug --gradients mnist.neural

🚀 Detect vanishing/exploding gradients with interactive charts.

🔹 3️⃣ Identify Dead Neurons

python neural.py debug --dead-neurons mnist.neural

🛠 Find layers with inactive neurons (common in ReLU networks).

🔹 4️⃣ Detect Training Anomalies

python neural.py debug --anomalies mnist.neural

🔥 Flag NaNs, weight explosions, and extreme activations.

🔹 5️⃣ Step Debugging (Interactive Tensor Inspection)

python neural.py debug --step mnist.neural

🔍 Pause execution at any layer and inspect tensors manually.


🌟 Why Neural?

Feature Neural Raw TensorFlow/PyTorch
Shape Validation ✅ Auto ❌ Manual
Framework Switching 1-line flag Days of rewriting
Architecture Diagrams Built-in Third-party tools
Training Config Unified Fragmented configs

🔄 Cross-Framework Code Generation

Neural DSL TensorFlow Output PyTorch Output
Conv2D(filters=32) tf.keras.layers.Conv2D(32) nn.Conv2d(in_channels, 32)
Dense(units=128) tf.keras.layers.Dense(128) nn.Linear(in_features, 128)

🏆 Benchmarks

Task Neural Baseline (TF/PyTorch)
MNIST Training 1.2x ⚡ 1.0x
Debugging Setup 5min 🕒 2hr+

📚 Documentation

Explore advanced features:

📚 Examples

Explore common use cases in examples/ with step-by-step guides in docs/examples/:

🕸Architecture Graphs (Zoom Alot For Some🫩)

classes packages


🤝 Contributing

We welcome contributions! See our:

To set up a development environment:

git clone https://github.com/yourusername/neural.git
cd neural
pip install -r requirements-dev.txt  # Includes linter, formatter, etc.
pre-commit install  # Auto-format code on commit

Star History

Star History Chart

Support

Please give us a star ⭐️ to increase our chances of getting into GitHub trends - the more attention we get, the higher our chances of actually making a difference. Please share this project with your friends! Every share helps us reach more developers and grow our community. The more developers we reach, the more likely we are to build something truly revolutionary together. 🚀

📬 Community

Note: This is an early release (v0.1.0) with known bugs—see GitHub for updates!

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

neural_dsl-0.2.4.tar.gz (79.9 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

neural_dsl-0.2.4-py3-none-any.whl (67.4 kB view details)

Uploaded Python 3

File details

Details for the file neural_dsl-0.2.4.tar.gz.

File metadata

  • Download URL: neural_dsl-0.2.4.tar.gz
  • Upload date:
  • Size: 79.9 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.12.3

File hashes

Hashes for neural_dsl-0.2.4.tar.gz
Algorithm Hash digest
SHA256 f29662e88d0a855708beb3650933f704bccb21b64d07b3a2aaf4edf6e451055a
MD5 102fed1e98df803deb5e3b5bf6d343f3
BLAKE2b-256 2212c53f9e02f109a75843bf93b498bb1d7f1c4bc30300dcbf36443f838df5a6

See more details on using hashes here.

File details

Details for the file neural_dsl-0.2.4-py3-none-any.whl.

File metadata

  • Download URL: neural_dsl-0.2.4-py3-none-any.whl
  • Upload date:
  • Size: 67.4 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.12.3

File hashes

Hashes for neural_dsl-0.2.4-py3-none-any.whl
Algorithm Hash digest
SHA256 dec62666e1bbf585e801eaab67018b769441e0c3432412714d785affab90dc70
MD5 79b5a3a5b6d478d535991b40da84dc84
BLAKE2b-256 54dc35b6d2b1d6ddbe53f45d872c773f3ee88668fab5436fa487eba5156a0c71

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page