Skip to main content
Ctrl+K
jaxpp - Home jaxpp - Home

jaxpp

jaxpp - Home jaxpp - Home

jaxpp

Table of Contents

Contents:

  • JaxPP
  • Installation
  • Usage
  • jaxpp
    • api
  • jaxpp

jaxpp#

  • api
    • BaseSchedule
    • Concat
    • DualPipeV
    • Eager1F1B
    • Interleaved1F1B
    • MpmdArray
      • is_mpmd_replicated
      • is_partially_addressable
      • sharding
      • to_mpmd_local_array
    • MpmdMesh
    • Std1F1B
    • ZeroBubble
    • collect_task_times_ms()
    • mpmd_to_spmd_reshard()
    • spmd_to_mpmd_reshard()
    • treduce()
    • treduce_i()

previous

Usage

next

api

NVIDIA NVIDIA
Privacy Policy | Manage My Privacy | Do Not Sell or Share My Data | Terms of Service | Accessibility | Corporate Policies | Product Security | Contact

Copyright © 2022-2026, NVIDIA CORPORATION. All rights reserved.