weight-sharing topic

List weight-sharing repositories

modules

43
Stars
9
Forks
Watchers

The official repository for our paper "Are Neural Nets Modular? Inspecting Functional Modularity Through Differentiable Weight Masks". We develop a method for analyzing emerging functional modularity...

param-share-transformer

24
Stars
4
Forks
Watchers

PyTorch implementation of Lessons on Parameter Sharing across Layers in Transformers

deeperwin

44
Stars
6
Forks
Watchers

DeepErwin is a python 3.8+ package that implements and optimizes JAX 2.x wave function models for numerical solutions to the multi-electron Schrödinger equation. DeepErwin supports weight-sharing when...