Python _split_index_for_jit - 2 примера найдено. Это лучшие примеры Python кода для jax._src.numpy.lax_numpy._split_index_for_jit, полученные из open source проектов. Вы можете ставить оценку каждому примеру, чтобы помочь нам улучшить качество примеров.