Метка: JAX

mathematical-operations-for-jax-pytrees

Mathematical operations for JAX pytrees

advantage-actor-critic-(a2c):-jax-+-flax-implementation

Advantage Actor Critic (A2C): jax + flax implementation

this-repository-implements-ppo-in-jax

This repository implements PPO in JAX

a-demo-of-how-to-use-jax-to-create-a-simple-gravity-simulation

A demo of how to use JAX to create a simple gravity simulation

a-framework-agnostic-trainer-interface-for-the-jax-ecosystem

A framework-agnostic Trainer interface for the Jax ecosystem

jax-based-neural-network-library

JAX-based neural network library