Vijay Janapa Reddi fb753882ec fix(module-01): Fix batched matmul and transpose grad preservation ...

- Change np.dot to np.matmul for proper batched 3D tensor multiplication
- Add requires_grad preservation in transpose() operation
- Fixes attention mechanism gradient flow issues

Regression tests added in tests/regression/test_gradient_flow_fixes.py

2025-10-27 20:28:53 -04:00

..

tensor_dev.ipynb

fix(module-01): Fix batched matmul and transpose grad preservation

2025-10-27 20:28:53 -04:00

tensor_dev.py

fix(module-01): Fix batched matmul and transpose grad preservation

2025-10-27 20:28:53 -04:00