NillionNetwork/nada-by-example

Implement the MaxPool operation

開放

#16 建立於 2024年8月14日

 (0 則留言) (0 個反應) (0 位負責人)Python (10 個分叉)auto 404
enhancementhacktoberfesthelp wanted

倉庫指標

星標
 (11 顆星)
PR 合併指標
 (PR 指標待抓取)

描述

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.

貢獻者指南