Neural network APIΒΆ fair_forge.nn.beutel A method for fair representations. fair_forge.nn.utils Utility functions for neural networks in JAX.