”flax“ 的搜索结果

#游戏引擎# Flax Engine - 多平台 3D 游戏引擎

Flax is a neural network library for JAX that is designed for flexibility.

Flax Engine Documentation

Collection of example projects for Flax Engine

#编辑器#Old repository with C# Editor and C# API for creating games in Flax Engine

#计算机科学#Gaussian processes in JAX and Flax.

#计算机科学#Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.

Implementing the Denoising Diffusion Probabilistic Model in Flax

general purpose programming language, in the vein of C++

GPT implementation in Flax

KoCLIP: Korean port of OpenAI CLIP, in Flax

#计算机科学#JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.

EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax

🤗 HuggingFace Diffusers Flax TPU and PyTorch GPU for Colab

Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`

#计算机科学#Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).

Pretraining Roberta on Marathi using TPU and JAX, Flax and Optax

#自然语言处理#Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.

Tutorial to pretrain & fine-tune a 🤗 Flax T5 model on a TPUv3-8 with GCP