Contents Menu Expand Light mode Dark mode Auto light/dark, in light mode Auto light/dark, in dark mode Skip to content
Taktiny
Logo

Getting Started

  • Installation
  • Quickstart

Maybe Useful

  • Saving and loading models
  • Parameter-Efficient Fine-Tuning (PEFT)
  • Data Loading & Transforms
  • Trainer & Callbacks
  • SPMD & Distributed Sharding

Tutorials

  • Linear Regression
  • Image Classification
  • Generative Adversarial Networks (GAN)

API Reference

  • taktiny.nn
    • Base Classes
    • Linear Layers
    • Activation Functions
    • Convolution & Pooling
    • Embedding
    • Normalization
    • Recurrent Layers
    • Transformer
    • Containers
    • Flatten & Unflatten
    • Regularization
    • Resampling
  • taktiny.takt
    • Takt API
    • PEFT Adapters
  • taktiny.data
    • Data Loading
    • Transforms & Packing
  • taktiny.trainer
    • Trainer
    • Configuration
    • Callbacks
  • taktiny.utils
    • SPMD Sharding
    • Quantization
    • Functional Transforms
    • Formatting
Back to top
View this page
Edit this page

taktiny.utilsΒΆ

Utilities for SPMD sharding, post-training quantization, functional transformations, and formatting.

  • SPMD Sharding
  • Quantization
  • Functional Transforms
  • Formatting
Next
SPMD Sharding
Previous
Callbacks
Copyright © 2026, Shinapri
Made with Sphinx and @pradyunsg's Furo