cargo / baracuda-kernels

Unified ML op facade for the baracuda CUDA ecosystem. Exposes every primitive an ML framework would expect (union of PyTorch torch.* + nn.functional and JAX lax.* / numpy ops) through a single Plan-based Rust surface, internally dispatching to baracuda-cutlass, the baracuda-* NVIDIA-library wrappers, or bespoke baracuda-kernels-sys kernels.

0.0.1-alpha.86 did not build; the badge stands for .

1,272 downloads · source · home page · cargo

cuda gpu kernel ml pytorch