v1.4.0-rc2

# What's new?

• (Added) Documentation is finally here!

---

# What's changed?

• (Changes) Several small fixes & improvements
  • Fixes incorrectness to dump & load State to optimizer & trainer.
  • Corrected Softmax backward pass to use the efficient Jacobian-vector product.
  • Added strict assertion to shapes for Math operations.
  • Softmax and Activations now reuse internal buffers to reduce memory allocation during training loops.
  • Fused Kernels: NN.Linear now combines matrix multiplication and bias addition into a single parallel block, reducing thread synchronization overhead significantly.