All modules for which code is available

• jax.numpy
• numpy
• ran.datasets.generate_cdl_channels
• ran.datasets.sionna_cdl_config
• ran.datasets.sionna_channel_dataset
• ran.mlir_trt_wrapper.compile
• ran.mlir_trt_wrapper.execute
• ran.mlir_trt_wrapper.jax_register_custom_call
• ran.mlir_trt_wrapper.mlir_trt_compiler_wrapper
• ran.mlir_trt_wrapper.mlir_trt_types
• ran.mlir_trt_wrapper.transform_mlir_custom_call_to_trt_plugin
• ran.phy.jax.pusch.pusch_inner_receiver
• ran.phy.jax.utils.awgn
• ran.phy.jax.utils.complex_ops
• ran.phy.numpy.pusch.channel_estimation
• ran.phy.numpy.pusch.codeblock_concatenation
• ran.phy.numpy.pusch.crc_decoder
• ran.phy.numpy.pusch.derate_matcher
• ran.phy.numpy.pusch.descrambler
• ran.phy.numpy.pusch.dmrs_utils
• ran.phy.numpy.pusch.equalizer
• ran.phy.numpy.pusch.gen_dmrs_sym
• ran.phy.numpy.pusch.ldpc_decoder
• ran.phy.numpy.pusch.measure_rssi
• ran.phy.numpy.pusch.noise_estimation
• ran.phy.numpy.pusch.noisevar_rsrp_sinr
• ran.phy.numpy.pusch.outer_receiver
• ran.phy.numpy.pusch.post_eq_noisevar_sinr
• ran.phy.numpy.pusch.pusch_rx
• ran.phy.numpy.pusch.soft_demapper
• ran.phy.numpy.utils.awgn
• ran.phy.numpy.utils.modulation
• ran.phy.numpy.utils.occ
• ran.phy.numpy.utils.sequence
• ran.trt_plugins.manager.global_trt_plugin_manager
• ran.trt_plugins.manager.inspect_trt_engine
• ran.trt_plugins.manager.trt_plugin_manager
• ran.utils._hdf5_load
• ran.utils._hdf5_save
• ran.utils._max_abs_diff
• ran.utils.config
• ran.utils.db