Subject
2 entries
Jax
Bookmarks
Ivy: The Unified Machine Learning Framework
Ivy is a unified machine learning framework that transpiles code between NumPy, PyTorch, TensorFlow, and JAX — write once, run anywhere across ML backends. An ambitious attempt to solve the framework fragmentation problem that plagues ML research and production.
Introduction to Graph Neural Networks with JAX/jraph
DeepMind's interactive introduction to Graph Neural Networks using JAX and jraph — covers message passing, graph classification, and node prediction with runnable code. One of the clearest practical GNN tutorials available as a Google Colab notebook.