[v0.2.7] Full API Review: consistency and naming conventions

[m96-chan]

Summary

Review all public APIs for consistency and naming conventions before v0.3.

Tasks

• Review pygpukit module exports
• Review GPUArray method names (snake_case consistency)
• Review scheduler module API
• Review llm module API
• Ensure consistent parameter ordering (shape, dtype pattern)
• Document any breaking changes

Motivation

Before v0.3 adds more features (Triton, advanced ops), establish stable API conventions.

References

• README.md v0.2.7 goals: "Full API review"

[m96-chan]

Completed in PR #76

API Exports Added

• ops module - Advanced operations access
• transpose - GPU matrix transpose
• bias_add_inplace - In-place bias addition
• linear_bias_gelu - Fused linear+bias+gelu

Public API (__all__)

__all__ = [
    # Core
    "GPUArray", "from_numpy", "zeros", "ones", "empty",
    # Ops
    "ops", "matmul", "add", "mul", "gelu", "layernorm",
    "transpose", "bias_add_inplace", "linear_bias_gelu",
    # JIT
    "cuda_jit",
    # LLM
    "llm",
    # Scheduler
    "scheduler",
]

Files Changed

• src/pygpukit/__init__.py - Added missing exports to __all__