Spark Optimization — Agent Skill by Seth Hobson

From the author

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.

— the description by Seth Hobson

AutoGineer summary

Written by AutoGineer from the skill’s own files, not by the author

This skill provides a comprehensive guide to optimizing Apache Spark applications. It covers essential performance factors including shuffle management, data skew mitigation, memory tuning, and effective partitioning strategies. The content includes both high-level architectural best practices and specific code patterns for common Spark challenges like join optimization, caching, and serialization.

What it does

Provides configuration patterns for Adaptive Query Execution (AQE).
Explains techniques for mitigating data skew through salting and broadcast joins.
Offers guidance on right-sizing partitions to balance parallelism and overhead.
Details strategies for effective memory management and persistence.
Outlines best practices for using columnar file formats like Parquet.
Demonstrates how to optimize joins using bucketed tables and broadcast hints.

When to use it

When debugging slow or failing Spark jobs.
When scaling data pipelines to handle larger datasets.
When trying to reduce network and disk I/O caused by excessive shuffles.
When optimizing resource usage and reducing executor memory pressure.
When preparing Spark jobs for production deployment.

Advantages

Reduces job execution time by minimizing expensive shuffle operations.
Improves cluster resource utilization through better memory and partition management.
Provides actionable patterns for handling common performance bottlenecks like data skew.
Encourages the use of modern Spark features like AQE for automatic optimization.

Usage examples

“How can I optimize a join between a large table and a small lookup table in Spark?”: The skill provides a pattern using the broadcast join hint to avoid a full shuffle, significantly speeding up the join operation.
“My Spark job is failing due to data skew on a specific key. What should I do?”: The skill suggests enabling Adaptive Query Execution (AQE) skew join optimization or implementing manual salting to distribute the skewed keys evenly.
“What is the recommended way to manage partitions for a large dataset?”: The skill explains how to calculate optimal partition counts based on data size and how to use repartitioning or coalescing to maintain efficiency.

Install

Download the skill as a ZIP:
https://mp-staging.autogineer.com/agent-skills/wshobson/spark-optimization/download

Unzip it into the cross-client skills folder of your project:

mkdir -p .agents/skills && unzip spark-optimization.zip -d .agents/skills

Documentation by Seth Hobson

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)

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

from pyspark.sql import SparkSession
from pyspark.sql import functions as F

# Create optimized Spark session
spark = (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 settings
df = (spark.read
    .format("parquet")
    .option("mergeSchema", "false")
    .load("s3://bucket/data/"))

# Efficient transformations
result = (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()

Files

ATTRIBUTION.mdAdded by AutoGineer986 B
LICENSELicence file added by AutoGineer1.0 KiB
SKILL.md3.1 KiB
references/details.md9.6 KiB

Requirements and compatibility

Works with: Claude Code, Claude apps, OpenAI Codex, GitHub Copilot, Cursor, Gemini CLI, Kiro, OpenCode, Goose, Amp, Roo Code, JetBrains Junie, Factory Droid, Devin (ex-Windsurf), Mistral Vibe, OpenHands, Letta

“Works with” means the tool reads the open Agent Skills format; a skill written for one tool may use features others lack.

Needs: Needs Python

Security

No issues found by our static scan· No issues found

A static scan is not a guarantee. Skills are third-party content: review a skill before you let an agent use it.

Anndata
K-Dense· Community
A Python framework for managing annotated data matrices in single-cell genomics and general research workflows.
Data and AIClaude CodeClaude appsOpenAI Codex+14 more
MITLow riskGitHub stars of the source repository56.4k
Liteparse
K-Dense· Community
Local document and PDF parsing with spatial text extraction and bounding box support.
Data and AIClaude CodeClaude appsOpenAI Codex+14 more
Apache-2.0Low riskHas scriptsGitHub stars of the source repository56.4k
Usfiscaldata
K-Dense· Community
Access U.S. Treasury federal financial data via the Fiscal Data REST API.
Data and AIClaude CodeClaude appsOpenAI Codex+14 more
MITLow riskGitHub stars of the source repository56.4k
Geopandas
K-Dense· Community
Guidance and local audit tools for Python workflows using GeoPandas 1.2.0 for spatial data analysis.
Data and AIClaude CodeClaude appsOpenAI Codex+14 more
MITLow riskHas scriptsGitHub stars of the source repository56.4k
Markitdown
K-Dense· Community
A lightweight utility for converting diverse document formats into structured Markdown for analysis, search, and LLM ingestion.
Data and AIClaude CodeClaude appsOpenAI Codex+14 more
MITLow riskHas scriptsGitHub stars of the source repository56.4k
Zarr Python
K-Dense· Community
A guide for storing and querying chunked N-D scientific arrays using Zarr-Python 3.
Data and AIClaude CodeClaude appsOpenAI Codex+14 more
MITLow riskGitHub stars of the source repository56.4k

Download ZIP

The ZIP holds the skill’s files, its licence files and an ATTRIBUTION.md. Download spark-optimization.zip