Skip to main content
Ctrl+K

IMLCV stable documentation

Getting started

  • Installing IMLCV
  • Contributions & Help
  • License
  • Authors
  • Changelog
  • Module Reference
    • IMLCV package
      • IMLCV.base package
      • IMLCV.configs package
      • IMLCV.examples package
      • IMLCV.external package
      • IMLCV.implementations package
        • IMLCV.implementations.tensorflow package
      • IMLCV.tools package
  • API Reference
    • IMLCV
      • IMLCV.base
        • IMLCV.base.CV
        • IMLCV.base.CVDiscovery
        • IMLCV.base.MdEngine
        • IMLCV.base.Observable
        • IMLCV.base.bias
        • IMLCV.base.rounds
      • IMLCV.configs
        • IMLCV.configs.bash_app_python
        • IMLCV.configs.config_general
        • IMLCV.configs.hpc_ugent
        • IMLCV.configs.local_htex
        • IMLCV.configs.local_threadpool
      • IMLCV.examples
        • IMLCV.examples.example_systems
        • IMLCV.examples.replot
        • IMLCV.examples.simul
      • IMLCV.external
        • IMLCV.external.tf2jax
      • IMLCV.implementations
        • IMLCV.implementations.tensorflow
        • IMLCV.implementations.CV
        • IMLCV.implementations.CvDiscovery
        • IMLCV.implementations.MdEngine
        • IMLCV.implementations.bias
        • IMLCV.implementations.energy
      • IMLCV.tools
        • IMLCV.tools._rbf_interp
        • IMLCV.tools._rbfinterp_pythran
        • IMLCV.tools.bessel_callback
        • IMLCV.tools.soap_kernel
        • IMLCV.tools.tools
      • IMLCV.scheme
  • .rst

IMLCV.external.tf2jax

Contents

  • Module Contents
    • Functions
      • loop_batcher()

IMLCV.external.tf2jax#

Module Contents#

Functions#

loop_batcher(prim, args, dims, **params)

IMLCV.external.tf2jax.loop_batcher(prim, args, dims, **params)[source]#
Contents
  • Module Contents
    • Functions
      • loop_batcher()

© Copyright 2023, David Devoogdt.