Skip to main content
Ctrl+K

StringJAX Tools 0.1.0 documentation

Start here

  • Usage Guide
  • Tutorial Notebooks
    • Automatic vectorisation
    • Manual vmap helpers
    • JIT helpers
    • Compilation-cache setup
    • Pytree policies
  • stringjax_tools
    • stringjax_tools.auto_vectorise
      • stringjax_tools.auto_vectorise.ArgSpec
      • stringjax_tools.auto_vectorise.auto_vmap
      • stringjax_tools.auto_vectorise.set_auto_vmap_defaults
      • stringjax_tools.auto_vectorise.get_auto_vmap_defaults
      • stringjax_tools.auto_vectorise.reset_auto_vmap_defaults
      • stringjax_tools.auto_vectorise.auto_vmap_defaults
      • stringjax_tools.auto_vectorise.set_auto_vmap_default_shapes
      • stringjax_tools.auto_vectorise.get_auto_vmap_default_shapes
      • stringjax_tools.auto_vectorise.reset_auto_vmap_default_shapes
      • stringjax_tools.auto_vectorise.auto_vmap_default_shapes
      • stringjax_tools.auto_vectorise.clear_auto_vmap_caches
    • stringjax_tools.vmap
      • stringjax_tools.vmap.vmapping_func
      • stringjax_tools.vmap.vmapping_func_cached
      • stringjax_tools.vmap.clear_vmap_caches
    • stringjax_tools.jit
      • stringjax_tools.jit.jit_with_static_args
      • stringjax_tools.jit.jit_with_dynamic_static_args
      • stringjax_tools.jit.is_static
    • stringjax_tools.cache
      • stringjax_tools.cache.configure_compilation_cache
    • stringjax_tools.pytrees
      • stringjax_tools.pytrees.PytreePolicy
      • stringjax_tools.pytrees.flatten_func
      • stringjax_tools.pytrees.unflatten_func_class
      • stringjax_tools.pytrees.make_pytree_flatteners

Python Module Index

s
 
s
- stringjax_tools
    stringjax_tools.auto_vectorise
    stringjax_tools.cache
    stringjax_tools.jit
    stringjax_tools.pytrees
    stringjax_tools.vmap

By Author name not set

© Copyright 2024-2026, Andreas Schachner.