weight-sharing topic

List weight-sharing repositories

modules

46
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

25
Stars
3
Forks
Watchers

PyTorch implementation of Lessons on Parameter Sharing across Layers in Transformers

deeperwin

47
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...