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

Index

B | C | D | E | I | J | M | S | T | Z

B

  • BaseSchedule (class in jaxpp.api)

C

  • collect_task_times_ms() (in module jaxpp.api)
  • Concat (class in jaxpp.api)

D

  • DualPipeV (class in jaxpp.api)

E

  • Eager1F1B (class in jaxpp.api)

I

  • Interleaved1F1B (class in jaxpp.api)
  • is_mpmd_replicated (jaxpp.api.MpmdArray property)
  • is_partially_addressable (jaxpp.api.MpmdArray property)

J

  • jaxpp.api
    • module

M

  • module
    • jaxpp.api
  • mpmd_to_spmd_reshard() (in module jaxpp.api)
  • MpmdArray (class in jaxpp.api)
  • MpmdMesh (class in jaxpp.api)

S

  • sharding (jaxpp.api.MpmdArray property)
  • spmd_to_mpmd_reshard() (in module jaxpp.api)
  • Std1F1B (class in jaxpp.api)

T

  • to_mpmd_local_array (jaxpp.api.MpmdArray property)
  • treduce() (in module jaxpp.api)
  • treduce_i() (in module jaxpp.api)

Z

  • ZeroBubble (class in jaxpp.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.