README.md
cli.py
pyproject.toml
pytensorforge.egg-info/PKG-INFO
pytensorforge.egg-info/SOURCES.txt
pytensorforge.egg-info/dependency_links.txt
pytensorforge.egg-info/entry_points.txt
pytensorforge.egg-info/requires.txt
pytensorforge.egg-info/top_level.txt
src/__init__.py
src/config.py
src/activations/Activation.py
src/activations/ELU.py
src/activations/GELU.py
src/activations/ReLU.py
src/activations/SELU.py
src/activations/Sigmoid.py
src/activations/Softmax.py
src/activations/Tanh.py
src/activations/__init__.py
src/core/Matrix.py
src/core/Scalar.py
src/core/Tensor.py
src/core/Vector.py
src/core/__init__.py
src/data/__init__.py
src/data/chat_dataset.py
src/data/corpus.py
src/data/document_stream.py
src/data/parallel_encode.py
src/data/prefetch.py
src/data/shard_builder.py
src/data/shard_writer.py
src/data/sharded_dataset.py
src/data/streaming_dataset.py
src/data/validation.py
src/inference/__init__.py
src/inference/chat_template.py
src/inference/config.py
src/inference/engine.py
src/inference/export.py
src/inference/kv_cache.py
src/inference/runtime.py
src/inference/sampling.py
src/inference/scheduler.py
src/inference/text.py
src/initializers/Constant.py
src/initializers/GlorotNormal.py
src/initializers/GlorotUniform.py
src/initializers/HeNormal.py
src/initializers/HeUniform.py
src/initializers/Initializer.py
src/initializers/LecunNormal.py
src/initializers/LecunUniform.py
src/initializers/Ones.py
src/initializers/Orthogonal.py
src/initializers/RandomNormal.py
src/initializers/RandomUniform.py
src/initializers/Zeros.py
src/initializers/__init__.py
src/loss/CategoricalCrossEntropy.py
src/loss/CrossEntropyLoss.py
src/loss/CrossEntropyWithLogitsLoss.py
src/loss/Hinge.py
src/loss/Huber.py
src/loss/Loss.py
src/loss/MSE.py
src/loss/MSELoss.py
src/loss/SparseCategoricalCrossEntropy.py
src/loss/__init__.py
src/loss/bce.py
src/loss/mae.py
src/math/__init__.py
src/math/clip.py
src/math/exp.py
src/math/log.py
src/math/sigmoid.py
src/models/__init__.py
src/models/embedding/Embedding.py
src/models/embedding/__init__.py
src/models/gpt/__init__.py
src/models/gpt/attention.py
src/models/gpt/block.py
src/models/gpt/config.py
src/models/gpt/context.py
src/models/gpt/model.py
src/models/gpt/recompute.py
src/models/gpt/rope.py
src/models/regression/Linear.py
src/models/regression/Logistic.py
src/models/regression/__init__.py
src/models/seq/Sequential.py
src/models/seq/__init__.py
src/models/svm/__init__.py
src/models/tokenizer/BPETokenizer.py
src/models/tokenizer/__init__.py
src/models/transformers/Dropout.py
src/models/transformers/LastToken.py
src/models/transformers/LayerNorm.py
src/models/transformers/Linear.py
src/models/transformers/MultiHeadAttention.py
src/models/transformers/TransformerBlock.py
src/models/transformers/__init__.py
src/neural/Dense.py
src/neural/LSTM.py
src/neural/Layer.py
src/neural/Parameter.py
src/neural/RNN.py
src/neural/__init__.py
src/ops/__init__.py
src/ops/stack.py
src/optimizers/Adagrad.py
src/optimizers/Adam.py
src/optimizers/AdamW.py
src/optimizers/Batch.py
src/optimizers/Nesterov.py
src/optimizers/Optimizer.py
src/optimizers/RMSProp.py
src/optimizers/SGD.py
src/optimizers/SGDMomentum.py
src/optimizers/__init__.py
src/scaling/StandardScaler.py
src/scaling/__init__.py
src/serialization/__init__.py
src/serialization/checkpoint.py
src/serialization/modelio.py
src/serving/__init__.py
src/serving/app.py
src/serving/config.py
src/serving/errors.py
src/serving/http.py
src/serving/metrics.py
src/serving/model_server.py
src/serving/protocol.py
src/serving/security.py
src/serving/server.py
src/tokenization/__init__.py
src/tokenization/base.py
src/tokenization/bpe.py
src/tokenization/bytebpe.py
src/tokenization/registry.py
src/training/__init__.py
src/training/checkpoint_manager.py
src/training/experiment.py
src/training/losses.py
src/training/precision.py
src/training/profiler.py
src/training/scheduler.py
src/training/trainer.py