Skip to content

Add Nx CPU benchmarks and refresh docs#1

Merged
goodhamgupta merged 3 commits into
mainfrom
benchmarks-nx-cpu
Jan 21, 2026
Merged

Add Nx CPU benchmarks and refresh docs#1
goodhamgupta merged 3 commits into
mainfrom
benchmarks-nx-cpu

Conversation

@goodhamgupta

Copy link
Copy Markdown
Owner

Summary

  • add Nx CPU backend baselines (Torchx/EXLA) to benchmark generation, CSV schema, and plots; add a warmup to avoid first-run skew
  • update the benchmark runner to include Nx CPU backends when available
  • refresh the README with benchmark environment details and updated latency/speedup ranges, and note the CPU baseline
  • runtime safety tweaks: gate AVX2 SIMD to avoid illegal instructions, reduce variable-length doc copying, remove deprecated rustler init args, and make Nx a required dependency

Testing

  • mix test
  • OPENBLAS_NUM_THREADS=1 RAYON_NUM_THREADS=8 mix run bench/generate_plots.exs
  • uv run bench/plot_benchmarks.py

@goodhamgupta goodhamgupta merged commit 7c80992 into main Jan 21, 2026
3 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant