objax
Research accelerator
An open-source machine learning framework designed to simplify and accelerate research
771 stars
25 watching
77 forks
Language: Python
last commit: 10 months ago
Linked from 1 awesome list
n2cholas/awesome-jaxRelated projects:
| Repository | Description | Stars |
|---|---|---|
 roberttlange/gymnax | A framework that accelerates RL environment processes by leveraging JAX and GPU acceleration | 650 |
| google/evojax | A toolkit for accelerating neuroevolution with neural networks on accelerators | 843 |
 asappresearch/flambe | An ML framework for accelerating research and its integration into production workflows | 262 |
 google-research/dex-lang | A research language for array processing, exploring type systems and program transformations. | 1,590 |
| google/paxml | A framework for configuring and running machine learning experiments on top of Jax. | 457 |
| google-research/noisystudent | A semi-supervised learning method to improve the accuracy of machine learning models by using noisy teacher models and student models. | 753 |
| google/praxis | A library designed to be usable by other JAX-based ML projects, providing optimized layers for machine learning at scale. | 178 |
| google-research/deep_ope | A set of pre-trained reinforcement learning policies and benchmarking data for offline model selection in reinforcement learning. | 85 |
 zk-ml/research | Research on integrating machine learning with emergent runtimes to improve performance and security. | 22 |
 gomlx/gomlx | A high-performance machine learning framework built on just-in-time compilation and automatic differentiation. | 354 |
 google-deepmind/distrax | A library of probability distributions and bijectors with a focus on readability, extensibility, and compatibility with existing frameworks. | 536 |
 jimjam-slam/talk-runapp-quarto-2022 | An introduction to Quarto as a tool for creating beautiful documents with R, Python, Julia, and Observable | 20 |
| google-deepmind/functa | A repository containing code for a meta-learning experiment on image datasets | 149 |
 orgexyz/blockagi | An AI research agent built on top of LangChain and OpenAI, designed to conduct iterative, domain-specific research and provide detailed narrative reports. | 293 |
| google-deepmind/jaxline | Provides a Python-based framework for building distributed JAX training and evaluation experiments | 152 |