tensorflow_lattice/python/lattice_lib.py (11 lines):
	- line 106: # TODO: investigate if there is a way to avoid sorting twice.
	- line 192: # TODO: Rename and update usage.
	- line 409: # TODO: run benchmark and figure out whether it make sense to merge
	- line 619: # TODO: Add final projection for unimodality constraints.
	- line 809: # TODO: It is likely that this algorithm will work for all trapezoid
	- line 1107: # TODO: approach used to implement regluarizers is likely to be more
	- line 1786: # TODO: unstacking entire set of weights for the purpuse of projection
	- line 1828: # TODO: Test whether adding min/max capping to dykstra projection would
	- line 2282: # TODO: Determine partial ordering of features by dominance and
	- line 2523: # TODO: investigate whether eps should be bigger.
	- line 2556: # TODO: actually assert them.


tensorflow_lattice/python/estimators.py (8 lines):
	- line 100: # TODO: support multi dim inputs.
	- line 101: # TODO: support multi dim output.
	- line 102: # TODO: add linear layer regularizers.
	- line 103: # TODO: add examples in docs.
	- line 104: # TODO: make _REPEATED_PAIR_DISCOUNT_IN_CRYSTALS_SCORE config param
	- line 432: # TODO: If warmstarting, look for the previous ensemble file.
	- line 1338: # TODO: handle cyclic PWL layers.
	- line 1829: # TODO: add support for KFL in RTL Layer


tensorflow_lattice/python/lattice_layer.py (2 lines):
	- line 708: # TODO: figure out whether it should be used.
	- line 903: # TODO: come up with a better solution than separately applying


tensorflow_lattice/python/kronecker_factored_lattice_lib.py (2 lines):
	- line 314: # TODO: optimize for case where all dims are monotonic and we won't
	- line 420: # TODO: in the case of only one monotonic dimension, we only have to


tensorflow_lattice/python/premade.py (2 lines):
	- line 54: # TODO: add support for serialization and object scoping or annoations.
	- line 392: # TODO: add support for tf.map_fn and inputs of shape (B, ?, input_dim)


tensorflow_lattice/python/kronecker_factored_lattice_layer.py (2 lines):
	- line 43: # TODO: add support for different lattice_sizes for each input
	- line 307: # TODO: can we remove this now that we always project at every step?


tensorflow_lattice/python/parallel_combination_layer.py (1 line):
	- line 32: # TODO: Add support for calibrators with units > 1.


tensorflow_lattice/python/categorical_calibration_layer.py (1 line):
	- line 32: # TODO: implement variation/variance regularizer.


tensorflow_lattice/python/configs.py (1 line):
	- line 574: # TODO: add option for different pre-aggregation model (linear/ensemble)


tensorflow_lattice/python/utils.py (1 line):
	- line 23: # TODO: update library not to explicitly check if None so we can return


tensorflow_lattice/python/pwl_calibration_sonnet_module.py (1 line):
	- line 22: # TODO: Add built-in regularizers like laplacian, hessian, etc.