Skip to content

feat: add optional GPU acceleration using backend abstraction#112

Open
Shrey45np wants to merge 1 commit intokarpathy:masterfrom
Shrey45np:feat-gpu-backend
Open

feat: add optional GPU acceleration using backend abstraction#112
Shrey45np wants to merge 1 commit intokarpathy:masterfrom
Shrey45np:feat-gpu-backend

Conversation

@Shrey45np
Copy link

This PR adds an optional GPU backend to micrograd using a backend abstraction.

Key points:

  • Uses CuPy for GPU execution when available
  • Falls back to NumPy for CPU-only systems
  • Does not modify the core scalar autograd logic
  • Adds benchmarking and demo scripts for performance comparison

This is an experimental, opt-in performance enhancement intended for educational and exploratory use.

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