Tag

#jax

3 articles

Meet Open Dreamer: A JAX/Flax Reproduction of the Dreamer 4 World Model Pipeline, With the Full Training Recipe Published

Learn about Open Dreamer, an open-source AI system that predicts future events by learning how the world works, using advanced machine learning techniques.

Jul 2549

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 22106

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 1899