JAX Featured LSTM in JAX & Flax (Complete example with code and notebook) LSTMs are a class of neural networks used to solve sequence problems such as time series and natural language processing. The LSTMs maintain some internal state that is useful in solving these problems. LSTMs apply for loops to iterate over each time step. We can use functions from JAX and
TensorFlow Featured How to build TensorFlow models with the Keras Functional API (Examples, code, and notebook) The Keras Functional API provides a way to build flexible and complex neural networks in TensorFlow. The Functional API is used to design networks that are not linear. In this article, you will discover that the Keras Functional API is used to create networks that: * Are non-linear. * Share layers. * Have
TensorFlow Featured How to create custom training loops in Keras Training models in Keras is usually done using the fit method. However, you may want more control over the training process. To do that, you'll need to create a custom training loop. This involves setting up a custom function to compute the loss and gradient. This article will
TensorFlow Featured How to build CNN in TensorFlow(examples, code, and notebooks) In the artificial neural networks with TensorFlow article, we saw how to build deep learning models with TensorFlow and Keras. We covered various concepts that are foundational in training neural networks with TensorFlow. In that article, we used a Pandas DataFrame to build a classification model in Keras. This article
TensorFlow Featured How to build artificial neural networks with Keras and TensorFlow Building artificial neural networks with TensorFlow and Keras requires understanding some key concepts. After learning these concepts, you'll install TensorFlow and start designing neural networks. This article will cover the concepts you need to comprehend to build neural networks in TensorFlow and Keras. Without further ado, let'
JAX Featured How to use TensorBoard in JAX & Flax Tracking machine learning experiments makes understanding and visualizing the model's performance easy. It also makes it possible to spot any problems in the network. For example, you can quickly spot overfitting by looking at the training and validation charts. You can plot these charts using your favorite charts
JAX Featured How to load datasets in JAX with TensorFlow JAX doesn't ship with data loading utilities. This keeps JAX focused on providing a fast tool for building and training machine learning models. Loading data in JAX is done using either TensorFlow or PyTorch. In the Image classification with JAX & Flax tutorial, we saw how to load
JAX Featured JAX loss functions Loss functions are at the core of training machine learning. They can be used to identify how well the model is performing on a dataset. Poor performance leads to a very high loss, while a well-performing model will have a lower loss. Therefore, the choice of a loss function is
JAX Featured JAX (What it is and how to use it in Python) JAX is a Python library offering high performance in machine learning with XLA and Just In Time (JIT) compilation. Its API is similar to NumPy's with a few differences. JAX ships with functionalities that aim to improve and increase speed in machine learning research. These functionalities include: * Automatic
JAX Featured Distributed training with JAX & Flax Training models on accelerators with JAX and Flax differs slightly from training with CPU. For instance, the data needs to be replicated in the different devices when using multiple accelerators. After that, we need to execute the training on multiple devices and aggregate the results. Flax supports TPU and GPU
JAX Featured Image classification with JAX & Flax Flax is a neural network library for JAX. JAX is a Python library that provides high-performance computing in machine learning research. JAX provides an API similar to NumPy making it easy to adopt. JAX also includes other functionalities for improving machine learning research. They include: * Automatic differentiation. JAX supports forward
TensorFlow Featured Object detection with TensorFlow 2 Object detection API Building object detection and image segmentation models is slightly different from other models. Majorly because you have to use specialized models and prepare the data in a particular way. This article will examine how to perform object detection and image segmentation on a custom dataset using the TensorFlow 2 Object
