JAX 0.7 compatibility

Đang mở
  • Có gì không ổn?
#2,010 5 bình luận 6 reaction 0 người được giao Xem trên GitHub

Chưa có ai nhận issue này.

Đánh giá

Độ khó
3/5
Thời gian dự kiến
1-2 ngày
Mức phù hợp với người mới
45/100
Loại issue
Lỗi
Độ rõ ràng
Khá rõ ràng
Mức độ hoạt động
Đình trệ
Công nghệ
python, tensorflow
Lĩnh vực
backend, machine-learning

Hướng nghiên cứu

Bắt đầu trong tensorflow_probability/python/internal/backend/jax/ops.py, quanh dòng 681, nơi traceback cho thấy mapping JAX đã bị xóa đang được truy cập. Kiểm tra cách sử dụng JAX backend trong phần mã xung quanh và phạm vi tương thích hiện có, sau đó xác minh rằng TensorFlow Probability có thể được import và hoạt động với JAX 0.7 mà không gặp AttributeError này.

Do mô hình lập chỉ mục viết ra từ nội dung của issue.

Mô tả

File ".../venvserver/lib/python3.12/site-packages/tensorflow_probability/python/internal/backend/jax/ops.py", line 681, in <module>
    jax.interpreters.xla.pytype_aval_mappings[onp.ndarray])
...
AttributeError: jax.interpreters.xla.pytype_aval_mappings was deprecated in JAX v0.5.0 and removed in JAX v0.7.0. jax.core.pytype_aval_mappings can be used as a replacement in most cases.

Seems like TFP for JAX was using deprecated features that are removed in the recent JAX release v0.7.0

Ngôn ngữ chính
Jupyter Notebook
Star
4.4k
Fork
1.1k
Không có pull request nào được merge trong 30 ngày

Chuẩn bị môi trường

Bắt đầu từ đâu

  1. Đọc hết issue, rồi đọc hướng dẫn đóng góp của dự án.
  2. Bình luận trên issue rằng bạn sẽ nhận — tránh hai người làm cùng một việc.
  3. Fork repository và làm thay đổi trên một nhánh.
  4. Mở pull request có tham chiếu số hiệu của issue.

Issue khác của tensorflow/probability

Tất cả issue của tensorflow/probability

Issue tương tự

Thêm issue về Backend & API Design

Nhận issue mới trong hộp thư của bạn

Bản tóm tắt ngắn những issue GitHub phù hợp với người mới.