JAX

Решение для запуска моделей Hugging Face в JAX после удаления нативной поддержки: использование torchax для преобразования PyTorch-моделей, работа с JIT-компиляцией и Pytree.