← Torna al catalogo
G

GitHub

Differentiable sign/round/floor for PyTorch

4.5(10,000 recensioni)
Gratuito
Lanciato nel 2026

Sobre

PyTorch's sign, round, floor, and argmax break gradients. Existing fixes (STE, Gumbel-Softmax) require rewriting your model. SLL-Core is zero-intrusive: with sll.linearize(): y = torch.sign(x) # now differentiable! loss.backward() Only linearizes inside an ε-band around decision boundaries. Everywhere else stays exactly hard. Zero deployment overhead. pip install sll-core

Punti di forza

  • +Zero deployment overhead
  • +Não requer reescrita do modelo
  • +Suporte a gradientes

Punti deboli

  • Limitado a ε-band em torno de decisões de fronteira
  • Não claro se é adequado para todos os casos

Você também pode gostar