Tag

#jax

2 articles

Implementing Deep Q-Learning (DQN) from Scratch Using RLax JAX Haiku and Optax to Train a CartPole Reinforcement Learning Agent

Learn to build a Deep Q-Network (DQN) reinforcement learning agent from scratch using JAX, RLax, Haiku, and Optax to solve the CartPole environment.

Mar 2220

A Coding Guide to Implement Advanced Differential Equation Solvers, Stochastic Simulations, and Neural Ordinary Differential Equations Using Diffrax and JAX

Learn how Neural Ordinary Differential Equations (Neural ODEs) use mathematical tools to predict future events, and how libraries like Diffrax and JAX make it easier to build these models.

Mar 1813