mlx-bun 0.2.0 → 0.3.0

package/src/sampler.ts

@@ -11,7 +11,7 @@
 import { Dtype } from "./mlx/ffi";
 import { MlxArray } from "./mlx/array";
 import * as ops from "./mlx/ops";
-import { applyCurve, type CurveParams } from "./curve-sampler";
+import { applyCurve, type CurveParams } from "./lab/curve/curve-sampler";
 
 /** Resolved HLG sampling config (the user-facing knobs). The mid gain is NOT
  *  here — it folds from temperature (m = 1/T) when the sampler is built. */
@@ -82,11 +82,11 @@ export interface SamplerOptions {
   seed?: number;
   /** HLG tone-curve sampling. When enabled, replaces temperature's flat slope
    *  with the piecewise curve (temperature becomes the mid gain). Off/undefined
-   *  ⇒ the plain temperature path, unchanged. docs/design/hlg-sampling.md. */
+   *  ⇒ the plain temperature path, unchanged. docs/archive/hlg-sampling.md. */
   hlg?: HlgConfig;
   /** v2 log-prob transfer-curve sampler. When set, the drawn monotone curve
    *  REPLACES temperature+softmax entirely (stochastic, seeded) — see
-   *  src/curve-sampler.ts. Identity curve ≡ temperature 1. */
+   *  src/lab/curve/curve-sampler.ts. Identity curve ≡ temperature 1. */
   curve?: CurveParams;
 }
 
@@ -310,7 +310,7 @@ export function applyXtc(
 }
 
 /** HLG sampling parameters — a piecewise tone curve on the logprobs.
- *  See docs/design/hlg-sampling.md. Knobs are in nats (the logprob unit). */
+ *  See docs/archive/hlg-sampling.md. Knobs are in nats (the logprob unit). */
 export interface HlgParams {
   /** m — mid-region slope. Folds temperature: m = 1/temperature. */
   gain: number;