jax-js-nonconsuming (Fork)
    Preparing search index...

    Module @hamk-uas/jax-js-nonconsuming

    Namespaces

    lax
    nn
    numpy
    profiler
    random
    scipyLinalg
    scipySpecial
    tree

    Enumerations

    DType
    MemoryEffect

    Classes

    Array
    ClosedJaxpr
    Jaxpr
    JitProgram
    SymDim

    Interfaces

    AotLinearizeOptions
    AotLinearizeResult
    BackendCapabilities
    CacheSizes
    CapturedCode
    CapturedKernel
    CapturedProgram
    CapturedRoutine
    CapturedSubProgram
    EffectVerificationResult
    GpuTimingResult
    JitReport
    JitStepCounts
    JsTreeDef
    LeakReport
    PerformanceBeliefState
    PrimalArtifact
    PullbackArtifact
    ResidualPack
    SnapshotEntry

    Type Aliases

    ArrayLike
    CodeCaptureEntry
    ConvClass
    Device
    Dim
    JsTree
    OwnedFunction
    ScanPath

    Variables

    _setJvpRetainedHandoffObserver
    checkLeaks
    devices
    profileGpuDetailed

    Functions

    _fusedCacheCompileCount
    _lastConvClass
    _lastConvRewritten
    _lastForiRewritten
    _setCalibrationState
    _setVerifyEffects
    aotLinearize
    blockUntilReady
    calibrateGpu
    captureJitReport
    clearCaches
    defaultDevice
    devicePut
    formatJitReport
    getBackend
    getCacheSizes
    getWebGPUDevice
    grad
    hasSymbolicDims
    hessian
    init
    isSymbolicDim
    jacfwd
    jacrev
    jit
    jitCompile
    jvp
    linearize
    makeJaxpr
    profileGpu
    reverse
    scatterAdd
    setCodeCapture
    setDebug
    valueAndGrad
    verifyJaxprEffects
    vjp
    vmap
    withBatch

    References

    jacobian → jacrev
    Namespaces
    Enumerations
    Classes
    Interfaces
    Type Aliases
    Variables
    Functions
    References