Path	Lines of Code
tensorflow_federated/experimental/python/__init__.py	1
tensorflow_federated/experimental/python/core/__init__.py	1
tensorflow_federated/experimental/python/core/api/__init__.py	1
tensorflow_federated/experimental/python/core/api/computations.py	4
tensorflow_federated/experimental/python/core/impl/__init__.py	1
tensorflow_federated/experimental/python/core/impl/jax_context/__init__.py	1
tensorflow_federated/experimental/python/core/impl/jax_context/jax_computation_context.py	10
tensorflow_federated/experimental/python/core/impl/jax_context/jax_serialization.py	83
tensorflow_federated/experimental/python/core/impl/wrappers/__init__.py	1
tensorflow_federated/experimental/python/core/impl/wrappers/computation_wrapper_instances.py	16
tensorflow_federated/experimental/python/learning/__init__.py	1
tensorflow_federated/experimental/python/learning/jax_components.py	56