#crf #candle #pytorch

candle-crf

porting pytorch-crf to candle

10 releases

new 0.8.8 Apr 6, 2025
0.8.7 Feb 28, 2025
0.8.6 Nov 28, 2024
0.8.5 Oct 15, 2024
0.7.3 Jul 18, 2024

#392 in Machine learning

Download history 3/week @ 2025-02-05 6/week @ 2025-02-19 152/week @ 2025-02-26 8/week @ 2025-03-05 123/week @ 2025-04-02

125 downloads per month

Custom license

82KB
2K SLoC

Candle CRF

Candle CRF ports pytorch-crf to Huggingface Candle.

Initialization

  • Pytorch CRF

    crf = CRF(num_tags, batch_first=false)
    
  • Candle CRF

    let crf = CRF::new(num_tags, false, &candle_core::Device::Cpu).unwrap();
    

Forward

  • Pytorch CRF

    llh = crf(emissions, tags, mask)
    
  • Candle CRF

    let llh = crf
            .forward(&emissions, &tags, Some(&mask), Reduction::default())
            .unwrap();
    

Decode

  • Pytorch CRF

    best_tags = crf.decode(emissions, mask)
    
  • Candle CRF

    let best_tags = crf.decode(&emissions, Some(&mask)).unwrap();
    

Dependencies

~9–18MB
~262K SLoC