JAX Articles & Tutorials by Weights & Biases

Skip to main content

Clear Search

English

Training a Protein Language Model on Cloud TPU

Jan 03

Articles, JAX, Uniprot, Advanced, Health Care, Experiment, NLP

Writing a Training Loop in JAX and Flax

Soumik Rakshit, Saurav Maheshkar

Jul 29

Articles, Computer Vision, Object Detection, CIFAR10, CNN, JAX, Tutorial, Advanced

How To Create an Image Classification Model in JAX/Flax

Saurav Maheshkar

Jun 23

CIFAR10, Articles, Computer Vision, Tutorial, JAX

Implementing NeRF in JAX

Soumik Rakshit, Saurav Maheshkar

Apr 29

JAX, Articles, Computer Vision

Matrix Factorization from Scratch in JAX: Regularized SVD for Recommendation Systems

Feb 07

Advanced, Tabular, JAX, Experiment, Tutorial, Plots, Sweeps, MovieLens, Recommender Systems

Bad Global Minima Exist and SGD Can Reach Them

Gustavo Sutter P. Carvalho, João Araújo, João Pedro Rodrigues Mattos, João Marcos Cardoso da Silva, Bruno Gomes Coelho

Aug 23

Intermediate, Computer Vision, Classification, JAX, Research, ResNet, Github, Plots, CIFAR10, RC

Popular Topics

Task

GenAI Agents Evaluations MLOps Fine-tuning All

Framework / Integration

Keras PyTorch HuggingFace GPT OpenAI All

Domain

Computer Vision Domain Agnostic NLP LLM Reinforcement Learning All

Iterate on AI agents and models faster. Try Weights & Biases today.

Sign up Try W&B now