GRX brings PyTorch-style tensor operations to Ruby. Every arithmetic op, activation, and optimizer step runs through a native C library compiled with AVX2+FMA SIMD. Ruby is the interface — C does the work. Features: autograd, SGD/Adam optimizers, Linear/Sequential/Dropout/BatchNorm layers, MSE/BCE/CrossEntropy loss functions, Xavier and He weight init. Cross-platform: .so on Linux, .dylib on macOS, .dll on Windows.

Required Ruby Version

>= 3.0.0

Authors

Angel Gabriel Garcia Razo

Versions

  1. 0.1.0 May 12, 2026 (27.5 KB)

Pushed by

SHA 256 checksum