jax

Платформа машинного обучения для преобразования числовых функций. Она описывается как объединяющая модифицированную версию autograd и XLA от TensorFlow. Она разработана так, чтобы максимально точно следовать структуре и рабочему процессу numpy, и работает с различными существующими платформами, такими как TensorFlow и PyTorch.