objax
by google
AI summary
Research accelerator
An open-source machine learning framework designed to simplify and accelerate research
- stars
- 772
- forks
- 76
- watching
- 25
- awesome list
- 1
Similar projects
Found by comparing what the projects do, not just their names.
RL accelerator
A framework that accelerates RL environment processes by leveraging JAX and GPU acceleration
Neuroevolution accelerator
A toolkit for accelerating neuroevolution with neural networks on accelerators
Research framework
An ML framework for accelerating research and its integration into production workflows
Array processor
A research language for array processing, exploring type systems and program transformations.
ML framework
A framework for configuring and running machine learning experiments on top of Jax.
Learning booster
A semi-supervised learning method to improve the accuracy of machine learning models by using noisy teacher models and student models.
ML framework component
A library designed to be usable by other JAX-based ML projects, providing optimized layers for machine learning at scale.
RL benchmarks
Provides benchmarking policies and datasets for offline reinforcement learning
Machine Learning Engine
Research on integrating machine learning with emergent runtimes to improve performance and security.
gomlx/gomlx553
Machine learning framework
An accelerated machine learning framework for Go that provides tools and libraries for training, fine-tuning, and modifying machine learning models.
Prob dist lib
A library of probability distributions and bijectors with a focus on readability, extensibility, and compatibility with existing frameworks.
Documentation generator
An introduction to Quarto as a tool for creating beautiful documents with R, Python, Julia, and Observable
Meta learning experiment
A repository containing code for a meta-learning experiment on image datasets
Research Agent
An AI research agent built on top of LangChain and OpenAI, designed to conduct iterative, domain-specific research and provide detailed narrative reports.
Experiment framework
Provides a Python-based framework for building distributed JAX training and evaluation experiments
Featured in 1 awesome list
Each link jumps to the spot where the list mentions objax.
