Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/PersiaML/PERSIA
/ types & classes
Types & classes
217 in github.com/PersiaML/PERSIA
⨍
Functions
720
◇
Types & classes
217
↳
Endpoints
1
↓ 7 callers
Class
IDTypeFeature
:class:`IDTypeFeature` is a sparse matrix in `LIL <https://scipy-lectures.org/advanced/scipy_sparse/lil_matrix.html>`_ format which contains categ
persia/embedding/data.py:69
↓ 7 callers
Class
PersiaBatch
r"""The :class:`PersiaBatch` is the type of dataset used internally in Persia. It wraps the :class:`IDTypeFeature`, :class:`NonIDTypeFeature`, :cl
persia/embedding/data.py:279
↓ 4 callers
Class
Values
Borrowing iterator over values of the linked array.
rust/persia-embedding-holder/src/array_linked_list.rs:993
↓ 3 callers
Class
DataCtx
r"""Data context provides the communication functionality to data generator component. Used for sending a :class:`PersiaBatch` to the `nn worker`
persia/ctx.py:274
↓ 3 callers
Class
DataLoader
examples/src/adult-income/data_generator.py:10
↓ 3 callers
Class
Indices
rust/persia-embedding-holder/src/array_linked_list.rs:990
↓ 3 callers
Class
IntoIndexed
rust/persia-embedding-holder/src/array_linked_list.rs:1098
↓ 3 callers
Class
Label
:class:`Label` is the ``subclass`` of :class:`NdarrayDataBase` that you can add various datatype and multi-dimensional data. Example for :cla
persia/embedding/data.py:217
↓ 3 callers
Class
StreamingDataset
r"""Streaming dataset receives the :class:`PersiaBatch` from the upstream data flow sent by :class:`DataCtx`. In the implemented :meth:`.Stre
persia/data.py:97
↓ 2 callers
Class
Adagrad
rust/persia-common/src/optim.rs:246
↓ 2 callers
Class
Adam
rust/persia-common/src/optim.rs:99
↓ 2 callers
Class
BaseCtx
r"""Initializes a common context for other persia context, e.g. :class:`DataCtx`, :class:`EmbeddingCtx` and :class:`TrainCtx`. This class should n
persia/ctx.py:202
↓ 2 callers
Class
DNN
examples/src/adult-income/model.py:7
↓ 2 callers
Class
DataLoader
r"""Data loader will preprocess the data to the ``PersiaTrainingBatch``. The :class:`DataLoader` is a pipeline that preprocess the :class:`.Persi
persia/data.py:202
↓ 2 callers
Class
IDTypeFeatureRemoteRef
rust/persia-common/src/lib.rs:141
↓ 2 callers
Class
NdarrayDataBase
The :class:`NdarrayDataBase` is a data structure that supports various datatypes and multi-dimensional data. PERSIA needs to convert the :class:`N
persia/embedding/data.py:160
↓ 2 callers
Class
NonIDTypeFeature
The :class:`NonIDTypeFeature` is the ``subclass`` of :class:`NdarrayDataBase` that you can add various datatypes and multi-dimensional data.
persia/embedding/data.py:256
↓ 2 callers
Class
SGD
r"""A wrapper to config the embedding-server SGD optimizer.
persia/embedding/optim.py:18
↓ 1 callers
Class
Adagrad
r"""A wrapper to config the embedding-server Adagrad optimizer.
persia/embedding/optim.py:60
↓ 1 callers
Class
Backward
rust/persia-core/src/backward.rs:358
↓ 1 callers
Class
DDPOption
Implements an option to convert torch model to a DDP model. Current backend in :class:`DDPOption` only support `nccl` and `gloo`. You can set
persia/distributed.py:74
↓ 1 callers
Class
EmbeddingConfig
r"""Embedding hyperparameters, argument of :class:`.EmbeddingCtx`.
persia/embedding/__init__.py:4
↓ 1 callers
Class
EmbeddingCtx
r"""Provides the embedding-related functionality. :class:`EmbeddingCtx` can run offline test or online inference depending on different preprocess
persia/ctx.py:345
↓ 1 callers
Class
Forward
rust/persia-core/src/forward.rs:834
↓ 1 callers
Class
GradientBatch
rust/persia-core/src/backward.rs:53
↓ 1 callers
Class
IDTypeFeatureWithSingleID
The :class:`IDTypeFeatureWithSingleID` is a special format of :class:`IDTypeFeature` where there is only one id for each sample in the batch. :cla
persia/embedding/data.py:116
↓ 1 callers
Class
InferCtx
r"""Subclass of :class:`EmbeddingCtx` that provides the inference functionality without `nats-servers`. Example for :class:`InferCtx`: .. co
persia/ctx.py:1077
↓ 1 callers
Class
IntoIndices
rust/persia-embedding-holder/src/array_linked_list.rs:1099
↓ 1 callers
Class
IntoValues
Owning iterator over values of the linked array.
rust/persia-embedding-holder/src/array_linked_list.rs:1102
↓ 1 callers
Class
OptimizerBase
rust/persia-core/src/optim.rs:9
↓ 1 callers
Class
PersiaBatchDataChannel
rust/persia-core/src/utils.rs:102
↓ 1 callers
Class
PersiaCommonContext
rust/persia-core/src/lib.rs:191
↓ 1 callers
Class
PersiaServiceCtx
Launch the required processes to mock the distributed PERSIA environment. You can add the embedding config file path or a dict that contains the
persia/helper.py:125
↓ 1 callers
Class
RawEmbedding
rust/persia-core/src/forward.rs:50
↓ 1 callers
Class
SumEmbedding
rust/persia-core/src/forward.rs:45
↓ 1 callers
Class
TestDataset
examples/src/adult-income/train.py:27
↓ 1 callers
Class
TrainCtx
r"""Subclass of :class:`EmbeddingCtx` that implements a `backward` function to update the embeddings. Example for :class:`TrainCtx`: ..
persia/ctx.py:655
↓ 1 callers
Class
_Env
persia/env.py:16
↓ 1 callers
Class
levelFilter
r"""Log level filter. Arguments: level (int): filter log level. Only logs with level higher than ``level`` will be kept.
persia/logger.py:8
Enum
Action
k8s/src/bin/operator.rs:50
Class
AdagradConfig
rust/persia-common/src/optim.rs:33
Class
Adam
r"""A wrapper to config the embedding-server Adam optimizer.
persia/embedding/optim.py:35
Class
AdamConfig
rust/persia-common/src/optim.rs:19
Class
AdamPowerOfBetas
rust/persia-common/src/optim.rs:94
Class
AllEmbeddingServerClient
rust/persia-embedding-server/src/embedding_worker_service/mod.rs:139
Interface
Allocatable
rust/persia-core/src/cuda/resource_pool.rs:7
Class
ApplyRequest
k8s/src/bin/server.rs:24
Class
ArrayLinkedList
rust/persia-embedding-holder/src/array_linked_list.rs:206
Class
BackwardImpl
rust/persia-core/src/backward.rs:115
Class
BaguaDistributedOption
Implements an option to convert torch model to a bagua distributed model. Example for :class:`BaguaDistributedOption`: .. code-block:: pytho
persia/distributed.py:259
Class
BatchEmbedding
rust/others/persia-common-benchmark/benches/serialize_inf_request.rs:22
Class
BatchEmbeddingSmallVec
rust/others/persia-common-benchmark/benches/serialize_inf_request.rs:108
Class
BatchEmbeddingsRequest
rust/others/persia-common-benchmark/benches/serialize_inf_request.rs:15
Class
BatchEmbeddingsRequestSmallVec
rust/others/persia-common-benchmark/benches/serialize_inf_request.rs:101
Class
BoundedGammaInitialization
rust/persia-embedding-config/src/lib.rs:41
Class
BoundedNormalInitialization
rust/persia-embedding-config/src/lib.rs:66
Class
BoundedPoissonInitialization
rust/persia-embedding-config/src/lib.rs:54
Class
BoundedUniformInitialization
rust/persia-embedding-config/src/lib.rs:28
Enum
CPUStorage
rust/persia-core/src/tensor.rs:98
Class
ChannelPair
rust/persia-common/src/utils.rs:4
Class
CheckpointingConfig
rust/persia-embedding-config/src/lib.rs:344
Class
Cli
k8s/src/bin/server.rs:196
Class
Cli
k8s/src/bin/gencrd.rs:9
Class
Cli
rust/persia-embedding-server/src/bin/persia-embedding-parameter-server.rs:26
Class
Cli
rust/persia-embedding-server/src/bin/persia-embedding-worker.rs:26
Class
CompactBatchEmbedding
rust/others/persia-common-benchmark/benches/serialize_inf_request.rs:65
Class
CompactBatchEmbeddingsRequest
rust/others/persia-common-benchmark/benches/serialize_inf_request.rs:57
Class
CompactBatchEmbeddingsRequestWithoutIndices
rust/others/persia-common-benchmark/benches/serialize_inf_request.rs:252
Class
ContextData
k8s/src/bin/operator.rs:40
Class
CudaEventPtr
rust/persia-core/src/cuda/cuda_event_pool.rs:12
Class
CudaMallocPtr
rust/persia-core/src/cuda/cuda_memory_pool.rs:11
Class
CudaStreamPtr
rust/persia-core/src/cuda/cuda_stream_pool.rs:10
Class
DLDataType
rust/persia-core/src/dlpack.rs:45
Enum
DLDataTypeCode
rust/persia-core/src/dlpack.rs:30
Class
DLDevice
rust/persia-core/src/dlpack.rs:54
Enum
DLDeviceType
rust/persia-core/src/dlpack.rs:11
Class
DLManagedTensor
rust/persia-core/src/dlpack.rs:81
Class
DLTensor
rust/persia-core/src/dlpack.rs:62
Enum
DTypeImpl
rust/persia-core/src/tensor.rs:24
Class
DataLoaderSpec
k8s/src/crd.rs:503
Class
DataflowService
rust/persia-core/src/nats.rs:117
Class
Device
rust/persia-core/src/tensor.rs:177
Enum
DeviceType
rust/persia-core/src/tensor.rs:171
Class
DistributedBaseOption
Implements a common option to convert torch model to a distributed data parallel model, e.g. Bagua Distributed or pyTorch DDP. This class sho
persia/distributed.py:13
Class
Dtype
rust/persia-core/src/forward.rs:102
Class
Embedding
rust/persia-core/src/forward.rs:58
Class
EmbeddingBackwardPacket
rust/persia-core/src/backward.rs:108
Class
EmbeddingBatch
rust/persia-common/src/lib.rs:110
Class
EmbeddingConfig
rust/persia-embedding-config/src/lib.rs:554
Class
EmbeddingGradientBatch
rust/persia-common/src/grad.rs:45
Enum
EmbeddingImpl
rust/persia-core/src/forward.rs:39
Class
EmbeddingModelInfo
rust/persia-model-manager/src/lib.rs:57
Class
EmbeddingModelManager
rust/persia-model-manager/src/lib.rs:74
Enum
EmbeddingModelManagerError
rust/persia-model-manager/src/lib.rs:27
Enum
EmbeddingModelManagerStatus
rust/persia-model-manager/src/lib.rs:64
Class
EmbeddingMonitorInner
rust/persia-embedding-server/src/monitor.rs:29
Class
EmbeddingParameterNatsService
rust/persia-embedding-server/src/embedding_parameter_service/mod.rs:596
Class
EmbeddingParameterServerConfig
rust/persia-embedding-config/src/lib.rs:419
Enum
EmbeddingParameterServerError
rust/persia-embedding-server/src/embedding_parameter_service/mod.rs:82
Class
EmbeddingParameterServerSpec
k8s/src/crd.rs:472
next →
1–100 of 217, ranked by callers