Spark Optimization

by wshobson46891e7e60daNo licenseListed Oct 8, 2026Updated Oct 8, 2026

Optimize Apache Spark jobs with partitioning, caching, shuffle optimization, and memory tuning. Use when improving Spark performance, debugging slow jobs, or scaling data processing pipelines.

Instructions onlyData & Analytics Software Development

AI-generated overview

Guides optimization of Apache Spark jobs through partitioning, caching, shuffle reduction, and memory tuning.

What it does: Provides production patterns for tuning Apache Spark jobs, covering partitioning strategies, memory and executor configuration, shuffle reduction, and data-skew handling. It includes a quick-start PySpark session configuration example, a table of key performance factors, and do's and don'ts. Deeper pattern documentation is referenced in a separate details file.
When to use it: Use when improving Spark performance, debugging slow jobs, or scaling data processing pipelines. Also relevant for tuning memory and executor settings, implementing partitioning strategies, or reducing shuffle and data skew.
Requirements: No scripts are shipped; it is instructions only. Running the example code requires Apache Spark with PySpark, and the example reads and writes Parquet data on S3, which needs appropriate storage access.

Apache Spark Optimization

Production patterns for optimizing Apache Spark jobs including partitioning strategies, memory management, shuffle optimization, and performance tuning.

When to Use This Skill

Optimizing slow Spark jobs
Tuning memory and executor configuration
Implementing efficient partitioning strategies
Debugging Spark performance issues
Scaling Spark pipelines for large datasets
Reducing shuffle and data skew

Core Concepts

1. Spark Execution Model

Driver Program    ↓Job (triggered by action)    ↓Stages (separated by shuffles)    ↓Tasks (one per partition)

Driver Program    ↓Job (triggered by action)    ↓Stages (separated by shuffles)    ↓Tasks (one per partition)

2. Key Performance Factors

Factor	Impact	Solution
Shuffle	Network I/O, disk I/O	Minimize wide transformations
Data Skew	Uneven task duration	Salting, broadcast joins
Serialization	CPU overhead	Use Kryo, columnar formats
Memory	GC pressure, spills	Tune executor memory
Partitions	Parallelism	Right-size partitions

Quick Start

python

from pyspark.sql import SparkSessionfrom pyspark.sql import functions as F
# Create optimized Spark sessionspark = (SparkSession.builder    .appName("OptimizedJob")    .config("spark.sql.adaptive.enabled", "true")    .config("spark.sql.adaptive.coalescePartitions.enabled", "true")    .config("spark.sql.adaptive.skewJoin.enabled", "true")    .config("spark.serializer", "org.apache.spark.serializer.KryoSerializer")    .config("spark.sql.shuffle.partitions", "200")    .getOrCreate())
# Read with optimized settingsdf = (spark.read    .format("parquet")    .option("mergeSchema", "false")    .load("s3://bucket/data/"))
# Efficient transformationsresult = (df    .filter(F.col("date") >= "2024-01-01")    .select("id", "amount", "category")    .groupBy("category")    .agg(F.sum("amount").alias("total")))
result.write.mode("overwrite").parquet("s3://bucket/output/")

from pyspark.sql import SparkSessionfrom pyspark.sql import functions as F
# Create optimized Spark sessionspark = (SparkSession.builder    .appName("OptimizedJob")    .config("spark.sql.adaptive.enabled", "true")    .config("spark.sql.adaptive.coalescePartitions.enabled", "true")    .config("spark.sql.adaptive.skewJoin.enabled", "true")    .config("spark.serializer", "org.apache.spark.serializer.KryoSerializer")    .config("spark.sql.shuffle.partitions", "200")    .getOrCreate())
# Read with optimized settingsdf = (spark.read    .format("parquet")    .option("mergeSchema", "false")    .load("s3://bucket/data/"))
# Efficient transformationsresult = (df    .filter(F.col("date") >= "2024-01-01")    .select("id", "amount", "category")    .groupBy("category")    .agg(F.sum("amount").alias("total")))
result.write.mode("overwrite").parquet("s3://bucket/output/")

Detailed patterns and worked examples

Detailed pattern documentation lives in references/details.md. Read that file when the navigation tier above is insufficient.

Best Practices

Do's

Enable AQE - Adaptive query execution handles many issues
Use Parquet/Delta - Columnar formats with compression
Broadcast small tables - Avoid shuffle for small joins
Monitor Spark UI - Check for skew, spills, GC
Right-size partitions - 128MB - 256MB per partition

Don'ts

Don't collect large data - Keep data distributed
Don't use UDFs unnecessarily - Use built-in functions
Don't over-cache - Memory is limited
Don't ignore data skew - It dominates job time
Don't use .count() for existence - Use .take(1) or .isEmpty()

Source and attribution

Source:wshobson/agentsinplugins/data-engineering/skills/spark-optimizationat commit46891e7

License: No license

Content belongs to its original authors. SourceWeft indexes it from a public repository.

Report or request removal

Spark Optimization

Apache Spark Optimization

When to Use This Skill

Core Concepts

1. Spark Execution Model

2. Key Performance Factors

Quick Start

Detailed patterns and worked examples

Best Practices

Do's

Don'ts

Source and attribution

More from wshobson/agents

Uv Package Manager

Web Component Design

Python Type Safety

Visual Design Foundations

Responsive Design

Python Resource Management

More in Data & Analytics

Spanner Basics

Gke Cost Analysis

Datalineage Bigquery Asset Impact Analysis

Bigquery Troubleshooting

Bigquery Optimization

Bigquery Bigframes