Source Files¶

• dir include
  • dir tenzr
    • file apiutils.h
    • file array.h
    • file control_flow.h
    • file custom_vjp.h
    • file debug.h
    • file execution.h
    • file grad.h
    • dir apiutils
      • file apiutils.h
    • dir check
      • file check.h
    • dir common
      • file error.h
      • file export.h
      • file logging.h
      • file macros.h
      • file trace.h
    • dir debug
      • file debug.h
    • dir execution
      • file ast_hash.h
      • file collective.h
      • file execution_engine.h
      • file jit_cache.h
      • file stream.h
    • dir frontend
      • file array.h
      • file dtype.h
      • file layout.h
      • file memory_pool.h
      • file node.h
      • file node_traversal.h
      • file op_registry.h
      • file shape.h
      • file sharding.h
      • file tensor.h
      • file ufunc.h
    • dir io
      • file safetensors.h
    • dir modules
      • file checkpoint.h
      • file data.h
      • dir checkpoint
        • file checkpoint_manager.h
      • dir data
        • file data.h
        • file dataloader.h
        • file dataset.h
        • file sampler.h
      • dir nn
        • file activations.h
        • file conv2d.h
        • file dropout.h
        • file embedding.h
        • file linear.h
        • file loss.h
        • file module.h
        • file nn.h
        • file norm.h
        • file sequential.h
      • dir optim
        • file adam.h
        • file clip.h
        • file ema.h
        • file lion.h
        • file optim.h
        • file optimizer.h
        • file schedule.h
        • file sgd.h
        • file transform.h
      • dir profiler
        • file profiler.h
      • dir random
        • file generator.h
        • file prng_key.h
        • file random.h
      • file modules.h
      • file nn.h
      • file optim.h
      • file profiler.h
      • file random.h
    • dir numpy
      • file numpy.h
    • dir plax
      • file plax.h
    • dir runtime
      • file device_buffer.h
      • file hardware_manager.h
      • file pjrt_error.h
      • file pjrt_executable.h
      • file pjrt_profiler.h
    • dir scipy
      • file scipy.h
    • dir sharding
      • file mesh.h
      • file named_sharding.h
      • file partition_spec.h
      • file shardy.h
    • dir topology
      • file torus_model.h
    • dir txpr
      • file control_flow.h
      • file custom_op_deriv.h
      • file custom_vjp.h
      • file emitter.h
      • file grad.h
      • file jit.h
      • file jvp.h
      • file node_to_txpr.h
      • file tracer.h
      • file txpr.h
      • file vmap.h
    • file io.h
    • file jit.h
    • file jvp.h
    • file numpy.h
    • file plax.h
    • file scipy.h
    • file sharding.h
    • file tenzr.h
    • file txpr.h
    • file vmap.h
• dir src
  • dir apiutils
    • file apiutils.cc
  • dir check
    • file check.cc
  • dir common
    • file trace.cc
    • file version.cc
  • dir debug
    • file debug.cc
  • dir execution
    • file ast_hash.cc
    • file collective.cc
    • file execution_engine.cc
    • file jit_cache.cc
    • file stream.cc
  • dir frontend
    • file array.cc
    • file dtype.cc
    • file layout.cc
    • file memory_pool.cc
    • file node.cc
    • file node_traversal.cc
    • file op_registry.cc
    • file shape.cc
    • file sharding.cc
    • file ufunc.cc
  • dir io
    • file safetensors.cc
  • dir modules
    • dir checkpoint
      • file checkpoint_manager.cc
    • dir data
      • file dataloader.cc
      • file dataset.cc
      • file sampler.cc
    • dir nn
      • file activations.cc
      • file conv2d.cc
      • file dropout.cc
      • file embedding.cc
      • file linear.cc
      • file loss.cc
      • file norm.cc
      • file sequential.cc
    • dir optim
      • file adam.cc
      • file clip.cc
      • file ema.cc
      • file lion.cc
      • file optimizer.cc
      • file sgd.cc
    • dir random
      • file generator.cc
      • file random.cc
  • dir numpy
    • file numpy.cc
  • dir plax
    • file plax.cc
  • dir python
    • file array_bindings.cc
    • file bindings.cc
    • file bindings.h
    • file plax_bindings.cc
    • file profiler_bindings.cc
    • file random_bindings.cc
    • file scipy_bindings.cc
    • file sharding_bindings.cc
    • file transform_bindings.cc
  • dir runtime
    • file hardware_manager.cc
    • file pjrt_profiler.cc
  • dir scipy
    • file scipy.cc
  • dir sharding
    • file shardy.cc
  • dir topology
    • file torus_model.cc
  • dir txpr
    • file control_flow.cc
    • file custom_op_deriv.cc
    • file custom_vjp.cc
    • file emitter.cc
    • file grad.cc
    • file jit.cc
    • file jvp.cc
    • file node_to_txpr.cc
    • file tracer.cc
    • file txpr.cc
    • file vmap.cc