← all repositories

lucidrains/lambda-networks

A PyTorch implementation of LambdaNetworks, a novel attention-based deep learning architecture for efficient image recognition.

1.5k stars Python Computer VisionML Frameworks
lambda-networks
Velocity · 7d
+0.7
★ / day
Trend
steady
star history

This repository provides a PyTorch implementation of LambdaNetworks, a new approach to image recognition that achieves state-of-the-art performance on ImageNet with reduced computational cost. The core innovation is the lambda layer, which captures interactions between inputs and context by transforming contexts into linear functions (lambdas) and applying them to each input element. The implementation supports both global and localized context variants, and includes both PyTorch and TensorFlow/Keras versions.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.