NillionNetwork/nada-by-example

Implement the MaxPool operation

Aberta

#16 aberto em 14 de ago. de 2024

 (0 comentário) (0 reação) (0 responsável)Python (10 forks)auto 404
enhancementhacktoberfesthelp wanted

Métricas do repositório

Stars
 (11 estrelas)
Métricas de merge de PR
 (Métricas PR pendentes)

Description

Description

Much like PyTorch, our MPC AI/ML framework nada-ai implements reusable standard AI modules. Pooling operations are one of these modules. Average pooling is already implemented here but max pooling is not.

Use Case

Max pooling is a very common operation in AI models - particularly computer vision models.

Specific Requirements

Our work on Average pooling should already provide solid template to performing kernel-style operations. The bit that would need to change is the actual operation that the kernel performs (average -> max) which can be found here.

Guia do colaborador