forgellm-runtime-0.7.6
- forgellm-runtime 0.7.6
- Permalink
- Docs.rs crate page
- MIT
- Links
- Homepage
- Repository
- crates.io
- Source
- Owners
- sauravpanda
- Dependencies
- forgellm-frontend ^0.7.6 normal
  thiserror ^2 normal
  tokenizers ^0.21 normal
  tracing ^0.1 normal
  pretty_assertions ^1 dev
- Versions
- 87.32% of the crate is documented
Platform
- x86_64-unknown-linux-gnu
Feature flags

docs.rs

Rust

Skip to main content

argmax

forgellm_runtime0.7.6

In forgellm_runtime::sampling

forgellm_runtime::sampling

Function argmax

pub fn argmax(logits: &[f32]) -> usize

Expand description

Greedy sampling: return the token with the highest logit.