Arktan logo

Shumai (Meta) is a fast differentiable tensor library in JavaScript and TypeScript that allows for efficient computation on tensors and supports automatic differentiation.

It is built with Bun and Flashlight and is designed for software engineers and researchers working with JavaScript and TypeScript.


  • Efficient computation on tensors
  • Support for automatic differentiation

Use Cases

  • Creating datasets
  • Training small models
  • Advanced/fine-grained training/inference logic
  • Building applications

Suited For

  • Software engineers
  • Researchers