Skip to content

Andrew Fairless, Ph.D.

Data, Science, and Tinkering

Overview
Experience and Education
Publications
SHAP Tutorial
Understanding the Basics of Bayesian Linear Regression
Classifying Medicine
The Peanuts Project

Search for:

Search for:

What I Read: Learning JAX

Home/What I Learn/What I Read: Learning JAX

By BylineAndrew Fairless on January 31, 2024November 20, 2023

https://kidger.site/thoughts/torch2jax/

Learning JAX as a PyTorch developer
Patrick Kidger
Posted on November 9, 2023

“Assuming you already know PyTorch, this is what I’ve found that you need to know to get up to speed with JAX.”

Cat Links What I Learn Tag Links JAX machine learning neural network PyTorch Torch

Post navigation

What I Read: Nvidia, GPU gold rushPrev post

What I Read: Finetuning LLMs Using LoRANext post

Categories

Bayesian statistics Machine Learning Statistics What I Learn What I Make

Tags

artificial intelligence attention Bayesian chatbot cognition computer vision database data engineering data science deployment DevOps economics efficiency embedding generalization generative GPU graph healthcare image interpretability large language model latency linear algebra machine learning medicine MLOps monitoring natural language processing neural network neuroscience optimization pipeline probability Python recurrent reinforcement learning reproducibility scalability software engineering SQL statistics training transformer unit test

Copyright © 2024 Andrew Fairless, Ph.D.. All Rights Reserved. | Simple Persona by Catch Themes

Scroll Up